Upload train_unsloth.py with huggingface_hub
Browse files- train_unsloth.py +1 -1
train_unsloth.py
CHANGED
|
@@ -58,7 +58,7 @@ trackio.init(
|
|
| 58 |
|
| 59 |
trainer = SFTTrainer(
|
| 60 |
model=model,
|
| 61 |
-
|
| 62 |
train_dataset=train_ds,
|
| 63 |
eval_dataset=eval_ds,
|
| 64 |
args=SFTConfig(
|
|
|
|
| 58 |
|
| 59 |
trainer = SFTTrainer(
|
| 60 |
model=model,
|
| 61 |
+
processing_class=tokenizer,
|
| 62 |
train_dataset=train_ds,
|
| 63 |
eval_dataset=eval_ds,
|
| 64 |
args=SFTConfig(
|