Upload train_alizee_coder.py with huggingface_hub
Browse files- train_alizee_coder.py +1 -1
train_alizee_coder.py
CHANGED
|
@@ -118,7 +118,7 @@ trainer = SFTTrainer(
|
|
| 118 |
train_dataset=train_dataset,
|
| 119 |
eval_dataset=eval_dataset,
|
| 120 |
peft_config=lora_config,
|
| 121 |
-
|
| 122 |
)
|
| 123 |
|
| 124 |
print("="*50)
|
|
|
|
| 118 |
train_dataset=train_dataset,
|
| 119 |
eval_dataset=eval_dataset,
|
| 120 |
peft_config=lora_config,
|
| 121 |
+
tokenizer=tokenizer,
|
| 122 |
)
|
| 123 |
|
| 124 |
print("="*50)
|