Upload train_aviation.py with huggingface_hub
Browse files- train_aviation.py +1 -1
train_aviation.py
CHANGED
|
@@ -163,7 +163,7 @@ config = SFTConfig(
|
|
| 163 |
report_to="trackio",
|
| 164 |
project="aviation-qa-tuning",
|
| 165 |
run_name="mistral-14b-sft-v1",
|
| 166 |
-
|
| 167 |
dataset_kwargs={"add_special_tokens": False} # Let tokenizer handle chat template
|
| 168 |
)
|
| 169 |
|
|
|
|
| 163 |
report_to="trackio",
|
| 164 |
project="aviation-qa-tuning",
|
| 165 |
run_name="mistral-14b-sft-v1",
|
| 166 |
+
max_length=2048,
|
| 167 |
dataset_kwargs={"add_special_tokens": False} # Let tokenizer handle chat template
|
| 168 |
)
|
| 169 |
|