christian.schwarz/VibeVoice
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update vibevoice_tokenizer_processor.py

Browse Source
This commit is contained in:
Wenhui Wang
2025-12-09 10:16:55 +08:00
parent 04d19f8352
commit 73a9711d8e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
vibevoice/processor/vibevoice_tokenizer_processor.py
+1 -1
View File
@@ -292,7 +292,7 @@ class VibeVoiceTokenizerProcessor(FeatureExtractionMixin):
return audio_array return audio_array
elif file_ext == '.pt': elif file_ext == '.pt':
# PyTorch tensor file # PyTorch tensor file
audio_tensor = torch.load(audio_path, map_location='cpu').squeeze() audio_tensor = torch.load(audio_path, map_location='cpu', weights_only=True).squeeze()
if isinstance(audio_tensor, torch.Tensor): if isinstance(audio_tensor, torch.Tensor):
audio_array = audio_tensor.numpy() audio_array = audio_tensor.numpy()
else: else: