From 73a9711d8e97b73c2f4d7fdb0bcf897235d7f43a Mon Sep 17 00:00:00 2001
From: Wenhui Wang <46838421+wenhui0924@users.noreply.github.com>
Date: Tue, 9 Dec 2025 10:16:55 +0800
Subject: [PATCH] Update vibevoice_tokenizer_processor.py

---
 vibevoice/processor/vibevoice_tokenizer_processor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vibevoice/processor/vibevoice_tokenizer_processor.py b/vibevoice/processor/vibevoice_tokenizer_processor.py
index 0d854b7..19b5795 100644
--- a/vibevoice/processor/vibevoice_tokenizer_processor.py
+++ b/vibevoice/processor/vibevoice_tokenizer_processor.py
@@ -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: