Upload train.py with huggingface_hub

train.py

@@ -134,6 +134,7 @@ 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