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 |
)
|