Update app.py

app.py

@@ -107,11 +107,11 @@ trainer = SFTTrainer(
     train_dataset=dataset,
     dataset_text_field="text",
     max_seq_length=max_seq_length,
-    dataset_num_proc=20,
+    dataset_num_proc=2,
     packing=False,
     args=TrainingArguments(
-        per_device_train_batch_size=20,
-        gradient_accumulation_steps=20,
+        per_device_train_batch_size=14,
+        gradient_accumulation_steps=14,
         learning_rate=2e-4,
         fp16=not is_bfloat16_supported(),
         bf16=is_bfloat16_supported(),