HirCoir commited on
Commit
7024203
1 Parent(s): 25a58a8

Update download-model.py

Browse files
Files changed (1) hide show
  1. download-model.py +5 -4
download-model.py CHANGED
@@ -37,16 +37,17 @@ else:
37
  filename=archivo_con_mayor_epoch,
38
  token=token,
39
  local_dir=".",
40
- local_filename="model.ckpt"
41
  )
 
 
42
  print(f"Archivo {archivo_con_mayor_epoch} descargado y renombrado como model.ckpt.")
43
 
44
- # Descargar el archivo config.json desde el mismo repositorio
45
  ruta_archivo_config = hf_hub_download(
46
  repo_id=repo_id,
47
  filename="config.json",
48
  token=token,
49
  local_dir=".",
50
- local_filename="config.json"
51
  )
52
- print("Archivo config.json descargado.")
 
 
37
  filename=archivo_con_mayor_epoch,
38
  token=token,
39
  local_dir=".",
 
40
  )
41
+ # Renombrar el archivo descargado a "model.ckpt"
42
+ os.rename(ruta_archivo_ckpt, "model.ckpt")
43
  print(f"Archivo {archivo_con_mayor_epoch} descargado y renombrado como model.ckpt.")
44
 
45
+ # Descargar el archivo config.json desde el repositorio
46
  ruta_archivo_config = hf_hub_download(
47
  repo_id=repo_id,
48
  filename="config.json",
49
  token=token,
50
  local_dir=".",
 
51
  )
52
+
53
+ print(f"Archivo config.json descargado a la ruta {ruta_archivo_config}.")