Update app.py
Browse files
app.py
CHANGED
|
@@ -121,6 +121,7 @@ trainer = SFTTrainer(
|
|
| 121 |
weight_decay=0.01,
|
| 122 |
lr_scheduler_type="linear",
|
| 123 |
seed=3407,
|
|
|
|
| 124 |
output_dir="outputs",
|
| 125 |
),
|
| 126 |
)
|
|
|
|
| 121 |
weight_decay=0.01,
|
| 122 |
lr_scheduler_type="linear",
|
| 123 |
seed=3407,
|
| 124 |
+
local_rank=4,
|
| 125 |
output_dir="outputs",
|
| 126 |
),
|
| 127 |
)
|