Spaces:

dad1909
/

CyberCode

Paused

dad1909 commited on Aug 17, 2024

Commit

972ca17

verified ·

1 Parent(s): d4b1572

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -113,7 +113,7 @@ trainer = SFTTrainer(
     dataset_num_proc=2,
     packing=False,
     args=TrainingArguments(
-        per_device_train_batch_size=2,
         gradient_accumulation_steps=2,
         learning_rate=2e-4,
         fp16=not is_bfloat16_supported(),

     dataset_num_proc=2,
     packing=False,
     args=TrainingArguments(
+        per_device_train_batch_size=1,
         gradient_accumulation_steps=2,
         learning_rate=2e-4,
         fp16=not is_bfloat16_supported(),