Update download-model.py
Browse files- 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
|
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 |
-
|
|
|
|
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}.")
|