Update vibevoice_tokenizer_processor.py

2025-12-09 10:16:55 +08:00
parent 04d19f8352
commit 73a9711d8e
1 changed files with 1 additions and 1 deletions
@@ -292,7 +292,7 @@ class VibeVoiceTokenizerProcessor(FeatureExtractionMixin):
            return audio_array
        elif file_ext == '.pt':
            # 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):
                audio_array = audio_tensor.numpy()
            else: