Upload train.py with huggingface_hub

train.py

@@ -122,6 +122,7 @@ training_args = SFTConfig(
     fp16=False,
     max_steps=1000,
     report_to="wandb",  # Disable reporting to avoid wandb prompts
+    output_dir="./tmp/sft-model",
 )
 
 """## Initialize and run the SFT Trainer"""
@@ -134,7 +135,6 @@ trainer = SFTTrainer(
     eval_dataset=dataset["test"] if "test" in dataset else None,
     peft_config=peft_config,
     processing_class=tokenizer,
-    output_dir="./tmp/sft-model",
 )
 
 # Train the model