Upload train_glm47_flash.py with huggingface_hub
Browse files- train_glm47_flash.py +1 -1
train_glm47_flash.py
CHANGED
|
@@ -200,7 +200,7 @@ trainer = SFTTrainer(
|
|
| 200 |
train_dataset=train_dataset,
|
| 201 |
eval_dataset=eval_dataset,
|
| 202 |
args=training_config,
|
| 203 |
-
|
| 204 |
peft_config=None, # Already applied above
|
| 205 |
)
|
| 206 |
|
|
|
|
| 200 |
train_dataset=train_dataset,
|
| 201 |
eval_dataset=eval_dataset,
|
| 202 |
args=training_config,
|
| 203 |
+
processing_class=tokenizer,
|
| 204 |
peft_config=None, # Already applied above
|
| 205 |
)
|
| 206 |
|