Update finetune_xtts_hf.py

finetune_xtts_hf.py

@@ -12,21 +12,21 @@ BASE_MODEL = "coqui/XTTS-v2"
 os.makedirs("./xtts_model", exist_ok=True)
 os.makedirs("./output", exist_ok=True)
 
-
-local_cache = "/tmp/hf_cache"
-
-os.makedirs(local_cache, exist_ok=True)
+# Cache seguro (en Spaces siempre se puede escribir en /tmp)
+HF_CACHE_DIR = "/tmp/hf_cache"
+os.makedirs(HF_CACHE_DIR, exist_ok=True)
 
 print("=== Descargando modelo base desde Hugging Face ===")
+
 model_dir = snapshot_download(
     repo_id="coqui/XTTS-v2",
-    local_dir="xtts_model",
-    cache_dir=local_cache,   # 👈 Forzamos el caché aquí
-    local_dir_use_symlinks=False,
-    resume_download=True
+    local_dir="/tmp/xtts_model",   # 👈 Descargar modelo en /tmp
+    cache_dir=HF_CACHE_DIR,        # 👈 Forzar caché también en /tmp
+    force_download=False,          # Descarga solo si no existe
 )
 
-print(f"Modelo descargado en: {model_dir}")
+print(f"✅ Modelo descargado en: {model_dir}")
+
 
 
 CONFIG_PATH = "./xtts_model/config.json"