Update app.py
Browse files
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=
|
| 114 |
-
gradient_accumulation_steps=
|
| 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(),
|