rahul7star commited on
Commit
1e1e4d4
·
verified ·
1 Parent(s): cab1df8

Update app_gpu.py

Browse files
Files changed (1) hide show
  1. app_gpu.py +1 -1
app_gpu.py CHANGED
@@ -282,7 +282,7 @@ def run_ui():
282
  logs = gr.Textbox(label="Logs (streaming)", lines=25)
283
 
284
  def launch(bm, ds, csv, sc, lc, out_dir, batch, num_w, r_, a_, ep_, lr_, max_rec, repo_):
285
- gen = train_lora_stream(
286
  bm, ds, csv, [sc, lc], out_dir,
287
  epochs=int(ep_), lr=float(lr_), r=int(r_), alpha=int(a_),
288
  batch_size=int(batch), num_workers=int(num_w),
 
282
  logs = gr.Textbox(label="Logs (streaming)", lines=25)
283
 
284
  def launch(bm, ds, csv, sc, lc, out_dir, batch, num_w, r_, a_, ep_, lr_, max_rec, repo_):
285
+ gen = train_lora_model(
286
  bm, ds, csv, [sc, lc], out_dir,
287
  epochs=int(ep_), lr=float(lr_), r=int(r_), alpha=int(a_),
288
  batch_size=int(batch), num_workers=int(num_w),