Upload train_code_reasoning.py

train_code_reasoning.py

@@ -166,7 +166,6 @@ training_args = SFTConfig(
     per_device_train_batch_size=4,
     gradient_accumulation_steps=4,
     learning_rate=5e-5,
-    max_seq_length=2048,
     warmup_ratio=0.03,
     lr_scheduler_type="cosine",
     bf16=True,
@@ -191,6 +190,7 @@ trainer = SFTTrainer(
     train_dataset=train_dataset,
     args=training_args,
     tokenizer=tokenizer,
+    max_seq_length=2048,
 )
 
 print("Starting training...")