Upload code/train_sft.py with huggingface_hub

code/train_sft.py

@@ -1109,6 +1109,8 @@ def main():
                 else:
                     repo_subfolder = f"sft_{args.version}"
 
+                api = HfApi(token=token)
+
                 if os.path.exists(save_path):
                     print(f"  ⬆️ Uploading final checkpoint {ckpt_filename}...")
                     api.upload_file(