Upload train_glm47_flash_test.py with huggingface_hub
Browse files
train_glm47_flash_test.py
CHANGED
|
@@ -142,7 +142,7 @@ trainer = SFTTrainer(
|
|
| 142 |
model=model,
|
| 143 |
train_dataset=dataset,
|
| 144 |
args=training_config,
|
| 145 |
-
|
| 146 |
)
|
| 147 |
|
| 148 |
print("\n" + "=" * 60)
|
|
|
|
| 142 |
model=model,
|
| 143 |
train_dataset=dataset,
|
| 144 |
args=training_config,
|
| 145 |
+
processing_class=tokenizer,
|
| 146 |
)
|
| 147 |
|
| 148 |
print("\n" + "=" * 60)
|