Upload train.py with huggingface_hub
Browse files
train.py
CHANGED
|
@@ -134,6 +134,7 @@ trainer = SFTTrainer(
|
|
| 134 |
eval_dataset=dataset["test"] if "test" in dataset else None,
|
| 135 |
peft_config=peft_config,
|
| 136 |
processing_class=tokenizer,
|
|
|
|
| 137 |
)
|
| 138 |
|
| 139 |
# Train the model
|
|
|
|
| 134 |
eval_dataset=dataset["test"] if "test" in dataset else None,
|
| 135 |
peft_config=peft_config,
|
| 136 |
processing_class=tokenizer,
|
| 137 |
+
output_dir="./tmp/sft-model",
|
| 138 |
)
|
| 139 |
|
| 140 |
# Train the model
|