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