From 697a23fb990e92a0097a7a147a44bdb7bcc4f18d Mon Sep 17 00:00:00 2001
From: leonard <l.hackel@tu-berlin.de>
Date: Fri, 28 Jun 2024 12:27:53 +0200
Subject: [PATCH] fixing bug of old vs new model name
---
scripts/improve_hf_training.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/scripts/improve_hf_training.py b/scripts/improve_hf_training.py
index 58fa17a..7c02c17 100644
--- a/scripts/improve_hf_training.py
+++ b/scripts/improve_hf_training.py
@@ -143,7 +143,7 @@ def main(
print(f"New model improved the compare metric by {new_metric - compare_metric:.4f}")
upload_model_and_readme_to_hub(
model=model,
- model_name=model_name,
+ model_name=new_model_name,
hf_entity=hf_entity,
test_results=new_results[0],
hparams=hparams,
--
GitLab