313 initial part
Browse files
app.py
CHANGED
|
@@ -117,7 +117,7 @@ trainer = SFTTrainer(
|
|
| 117 |
bf16=is_bfloat16_supported(),
|
| 118 |
warmup_steps=5,
|
| 119 |
logging_steps=10,
|
| 120 |
-
max_steps=
|
| 121 |
optim="adamw_8bit",
|
| 122 |
weight_decay=0.01,
|
| 123 |
lr_scheduler_type="linear",
|
|
|
|
| 117 |
bf16=is_bfloat16_supported(),
|
| 118 |
warmup_steps=5,
|
| 119 |
logging_steps=10,
|
| 120 |
+
max_steps=313,
|
| 121 |
optim="adamw_8bit",
|
| 122 |
weight_decay=0.01,
|
| 123 |
lr_scheduler_type="linear",
|