Upload download_model_weight.py

download_model_weight.py

@@ -108,17 +108,17 @@ def main():
     model_config = {
         "sft": {
             "repo_id": "YuvrajSingh9886/smol-llama-finetuned",
-            "filename": "fine_tuned_model.pt",
+            "filename": "snapshot_fine_tuned_model_900.pt",
             "cache_dir": "weights/fine_tuned"
         },
         "dpo": {
             "repo_id": "YuvrajSingh9886/smol-llama-dpo",
-            "filename": "dpo_model.pt",
+            "filename": "DPO_model_1650.pt",
             "cache_dir": "weights/DPO"
         },
         "pretrained": {
             "repo_id": "YuvrajSingh9886/smol-llama-base",
-            "filename": "pretrained_model.pt",
+            "filename": "snapshot_6750.pt",
             "cache_dir": "weights/pretrained"
         }
     }