Update finetune_xtts_hf.py

finetune_xtts_hf.py

@@ -9,6 +9,9 @@ DATASET_PATH = "/home/user/app/dataset"        # Ruta a tu dataset
 OUTPUT_PATH = "/tmp/output_model"
 BASE_MODEL = "coqui/XTTS-v2"
 
+os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1"
+os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "0"
+
 os.makedirs("/tmp/xtts_cache", exist_ok=True)
 os.chmod("/tmp/xtts_cache", 0o777)
 
@@ -18,11 +21,18 @@ os.chmod("/tmp/xtts_model", 0o777)
 # Continúa con tu lógica, usando las nuevas rutas de manera consistent
 
 # Base model download
+# model_dir = snapshot_download(
+#     repo_id="coqui/XTTS-v2",
+#     local_dir="/tmp/xtts_model",   # modelo temporal en /tmp
+#     cache_dir="/tmp/xtts_cache",        # cache seguro dentro de tu espacio
+#     force_download=False,
+)
 model_dir = snapshot_download(
-    repo_id="coqui/XTTS-v2",
-    local_dir="/tmp/xtts_model",   # modelo temporal en /tmp
-    cache_dir="/tmp/xtts_cache",        # cache seguro dentro de tu espacio
-    force_download=False,
+    "coqui/XTTS-v2",
+    cache_dir="/tmp/hf_models",
+    local_files_only=False,
+    resume_download=True,
+    use_auth_token=True
 )
 
 print(f"✅ Modelo descargado en: {model_dir}")