dad1909 commited on
Commit
c2ab1f3
·
verified ·
1 Parent(s): efe1848

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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
  )