Update app.py

app.py

@@ -110,14 +110,13 @@ trainer = SFTTrainer(
     dataset_num_proc=2,
     packing=False,
     args=TrainingArguments(
-        per_device_train_batch_size=5,
-        gradient_accumulation_steps=5,
+        per_device_train_batch_size=6,
+        gradient_accumulation_steps=6,
         learning_rate=2e-4,
         fp16=not is_bfloat16_supported(),
         bf16=is_bfloat16_supported(),
         warmup_steps=5,
         logging_steps=10,
-        max_steps=313,
         optim="adamw_8bit",
         weight_decay=0.01,
         lr_scheduler_type="linear",