Spaces:
Running
Running
Update app_gpu.py
Browse files- 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 =
|
| 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),
|