dad1909 commited on
Commit
b787ad1
·
verified ·
1 Parent(s): 203feb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -110,8 +110,8 @@ trainer = SFTTrainer(
110
  dataset_num_proc=2,
111
  packing=False,
112
  args=TrainingArguments(
113
- per_device_train_batch_size=14,
114
- gradient_accumulation_steps=14,
115
  learning_rate=2e-4,
116
  fp16=not is_bfloat16_supported(),
117
  bf16=is_bfloat16_supported(),
 
110
  dataset_num_proc=2,
111
  packing=False,
112
  args=TrainingArguments(
113
+ per_device_train_batch_size=5,
114
+ gradient_accumulation_steps=5,
115
  learning_rate=2e-4,
116
  fp16=not is_bfloat16_supported(),
117
  bf16=is_bfloat16_supported(),