Upload run_sft_job.py with huggingface_hub
Browse files- run_sft_job.py +1 -1
run_sft_job.py
CHANGED
|
@@ -144,7 +144,7 @@ trainer = SFTTrainer(
|
|
| 144 |
eval_dataset=eval_dataset,
|
| 145 |
peft_config=peft_config,
|
| 146 |
args=training_args,
|
| 147 |
-
|
| 148 |
dataset_text_field="messages", # Use the messages column
|
| 149 |
packing=False # Do not pack sequences
|
| 150 |
)
|
|
|
|
| 144 |
eval_dataset=eval_dataset,
|
| 145 |
peft_config=peft_config,
|
| 146 |
args=training_args,
|
| 147 |
+
max_length=2048,
|
| 148 |
dataset_text_field="messages", # Use the messages column
|
| 149 |
packing=False # Do not pack sequences
|
| 150 |
)
|