Spaces:
Running
on
Zero
Running
on
Zero
Add fallback mechanism for loading voicepack in TTSModel
Browse files- tts_model.py +5 -1
tts_model.py
CHANGED
@@ -193,7 +193,11 @@ class TTSModel:
|
|
193 |
for voice in voice_names:
|
194 |
try:
|
195 |
voice_path = os.path.join(self.voices_dir, "voices", f"{voice}.pt")
|
196 |
-
|
|
|
|
|
|
|
|
|
197 |
t_voices.append(voicepack)
|
198 |
except Exception as e:
|
199 |
print(f"Warning: Failed to load voice {voice}: {str(e)}")
|
|
|
193 |
for voice in voice_names:
|
194 |
try:
|
195 |
voice_path = os.path.join(self.voices_dir, "voices", f"{voice}.pt")
|
196 |
+
try:
|
197 |
+
voicepack = torch.load(voice_path, weights_only=True)
|
198 |
+
except Exception as e:
|
199 |
+
print(f"Warning: weights_only load failed, attempting full load: {str(e)}")
|
200 |
+
voicepack = torch.load(voice_path, weights_only=False)
|
201 |
t_voices.append(voicepack)
|
202 |
except Exception as e:
|
203 |
print(f"Warning: Failed to load voice {voice}: {str(e)}")
|