Update download-model.py

download-model.py

@@ -31,14 +31,22 @@ else:
     # Obtener el archivo con la mayor "epoch"
     archivo_con_mayor_epoch = max(archivos_ckpt, key=get_epoch)
 
-    # Descargar el archivo con la mayor "epoch" al directorio actual
-    ruta_archivo_descargado = hf_hub_download(
+    # Descargar el archivo con la mayor "epoch" como "model.ckpt"
+    ruta_archivo_ckpt = hf_hub_download(
         repo_id=repo_id,
         filename=archivo_con_mayor_epoch,
-        token=token
+        token=token,
+        local_dir=".",
+        local_filename="model.ckpt"
     )
-
-    # Renombrar el archivo descargado
-    os.rename(ruta_archivo_descargado, "model.ckpt")
-
     print(f"Archivo {archivo_con_mayor_epoch} descargado y renombrado como model.ckpt.")
+
+# Descargar el archivo config.json desde el mismo repositorio
+ruta_archivo_config = hf_hub_download(
+    repo_id=repo_id,
+    filename="config.json",
+    token=token,
+    local_dir=".",
+    local_filename="config.json"
+)
+print("Archivo config.json descargado.")