sob111 commited on
Commit
3f1a94d
·
verified ·
1 Parent(s): d7dd4e2

Update finetune_xtts_hf.py

Browse files
Files changed (1) hide show
  1. finetune_xtts_hf.py +7 -25
finetune_xtts_hf.py CHANGED
@@ -9,45 +9,27 @@ DATASET_PATH = "/home/user/app/dataset" # Ruta a tu dataset
9
  OUTPUT_PATH = "/tmp/output_model"
10
  BASE_MODEL = "coqui/XTTS-v2"
11
 
 
 
 
12
  os.makedirs("/tmp/xtts_model", exist_ok=True)
13
  os.chmod("/tmp/xtts_model", 0o777)
14
 
15
- # Cache seguro dentro de tu espacio
16
- hf_cache_dir = "/data/xtts_model_cache"
17
- os.makedirs(hf_cache_dir, exist_ok=True) # Crear carpeta si no existe
18
- # os.makedirs("/data/xtts_model", exist_ok=True) # Crear carpeta si no existe
19
-
20
- PERSISTENT_DATA_DIR = "/data/my_app_data"
21
-
22
- # Crea este directorio si no existe.
23
- # Esta es la única operación de 'makedirs' de alto nivel que necesitas.
24
- os.makedirs(PERSISTENT_DATA_DIR, exist_ok=True)
25
-
26
- # Ahora, define todos los demás directorios como subdirectorios de este.
27
- # Esto garantiza que todas las operaciones de escritura se realicen en un lugar con permisos.
28
- hf_cache_dir = os.path.join(PERSISTENT_DATA_DIR, "xtts_model_cache")
29
- model_download_dir = os.path.join(PERSISTENT_DATA_DIR, "xtts_model")
30
-
31
- # Ahora, puedes crear estos directorios sin problema.
32
- os.makedirs(hf_cache_dir, exist_ok=True)
33
- os.makedirs(model_download_dir, exist_ok=True)
34
-
35
  # Continúa con tu lógica, usando las nuevas rutas de manera consistent
36
 
37
  # Base model download
38
  model_dir = snapshot_download(
39
  repo_id="coqui/XTTS-v2",
40
- local_dir=model_download_dir, # modelo temporal en /tmp
41
- cache_dir=hf_cache_dir, # cache seguro dentro de tu espacio
42
  force_download=False,
43
  )
44
 
45
  print(f"✅ Modelo descargado en: {model_dir}")
46
 
47
 
48
-
49
- CONFIG_PATH = "/data/xtts_model/config.json"
50
- RESTORE_PATH = "/data/xtts_model/model.pth"
51
 
52
  # === 2. Editar configuración para tu dataset VoxPopuli ===
53
  print("=== Editando configuración para fine-tuning con VoxPopuli ===")
 
9
  OUTPUT_PATH = "/tmp/output_model"
10
  BASE_MODEL = "coqui/XTTS-v2"
11
 
12
+ os.makedirs("/tmp/xtts_cache", exist_ok=True)
13
+ os.chmod("/tmp/xtts_cache", 0o777)
14
+
15
  os.makedirs("/tmp/xtts_model", exist_ok=True)
16
  os.chmod("/tmp/xtts_model", 0o777)
17
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
  # Continúa con tu lógica, usando las nuevas rutas de manera consistent
19
 
20
  # Base model download
21
  model_dir = snapshot_download(
22
  repo_id="coqui/XTTS-v2",
23
+ local_dir="/tmp/xtts_model", # modelo temporal en /tmp
24
+ cache_dir="/tmp/xtts_cache", # cache seguro dentro de tu espacio
25
  force_download=False,
26
  )
27
 
28
  print(f"✅ Modelo descargado en: {model_dir}")
29
 
30
 
31
+ CONFIG_PATH = "/tmp/xtts_model/config.json"
32
+ RESTORE_PATH = "/tmp/xtts_model/model.pth"
 
33
 
34
  # === 2. Editar configuración para tu dataset VoxPopuli ===
35
  print("=== Editando configuración para fine-tuning con VoxPopuli ===")