Upload xtts_demo.py

xtts_demo.py

@@ -14,16 +14,30 @@ import numpy as np
 import torch
 import torchaudio
 import traceback
-from utils.formatter import format_audio_list,find_latest_best_model, list_audios
+from utils.formatter import format_audio_list, find_latest_best_model, list_audios
 from utils.gpt_train import train_gpt
 
 from faster_whisper import WhisperModel
 
 from TTS.tts.configs.xtts_config import XttsConfig
-from TTS.tts.models.xtts import Xtts
+from TTS.tts.configs.shared_configs import BaseTTSConfig
+from TTS.tts.models.xtts import Xtts, XttsAudioConfig, XttsArgs
+from TTS.config.shared_configs import BaseDatasetConfig
 
 import requests
 
+# ✅ PARCHE COMPLETO PyTorch 2.6+
+import torch.serialization
+torch.serialization.add_safe_globals([
+    XttsConfig,
+    BaseTTSConfig,
+    BaseDatasetConfig,
+    Xtts,
+    XttsAudioConfig,
+    XttsArgs  # ← NUEVA CLASE que pidió este último error
+])
+
+
 def download_file(url, destination):
     try:
         response = requests.get(url, stream=True)