sunkencity commited on
Commit
bfbfcaf
·
verified ·
1 Parent(s): fdda1e0

Upload train_aviation.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- max_seq_length=2048,
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