Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -567,10 +567,10 @@ def task_resize(hf_token, lora_input, new_rank, dynamic_method, dynamic_param, o
|
|
| 567 |
css = ".container { max-width: 900px; margin: auto; }"
|
| 568 |
|
| 569 |
with gr.Blocks() as demo:
|
| 570 |
-
gr.Markdown("# 🧰
|
| 571 |
|
| 572 |
with gr.Tabs():
|
| 573 |
-
with gr.Tab("Merge + Reshard"):
|
| 574 |
t1_token = gr.Textbox(label="Token", type="password")
|
| 575 |
t1_base = gr.Textbox(label="Base Repo", value="ostris/Z-Image-De-Turbo")
|
| 576 |
t1_sub = gr.Textbox(label="Subfolder", value="transformer")
|
|
@@ -586,7 +586,7 @@ with gr.Blocks() as demo:
|
|
| 586 |
t1_res = gr.Textbox(label="Result")
|
| 587 |
t1_btn.click(task_merge, [t1_token, t1_base, t1_sub, t1_lora, t1_scale, t1_prec, t1_shard, t1_out, t1_struct, t1_priv], t1_res)
|
| 588 |
|
| 589 |
-
with gr.Tab("Extract"):
|
| 590 |
t2_token = gr.Textbox(label="Token", type="password")
|
| 591 |
t2_org = gr.Textbox(label="Original")
|
| 592 |
t2_tun = gr.Textbox(label="Tuned")
|
|
@@ -596,7 +596,7 @@ with gr.Blocks() as demo:
|
|
| 596 |
t2_res = gr.Textbox(label="Result")
|
| 597 |
t2_btn.click(task_extract, [t2_token, t2_org, t2_tun, t2_rank, t2_out], t2_res)
|
| 598 |
|
| 599 |
-
with gr.Tab("Merge Adapters
|
| 600 |
t3_token = gr.Textbox(label="Token", type="password")
|
| 601 |
t3_urls = gr.Textbox(label="URLs")
|
| 602 |
with gr.Row():
|
|
@@ -607,7 +607,7 @@ with gr.Blocks() as demo:
|
|
| 607 |
t3_res = gr.Textbox(label="Result")
|
| 608 |
t3_btn.click(task_merge_adapters, [t3_token, t3_urls, t3_beta, t3_sigma, t3_out], t3_res)
|
| 609 |
|
| 610 |
-
with gr.Tab("Resize"):
|
| 611 |
t4_token = gr.Textbox(label="Token", type="password")
|
| 612 |
t4_in = gr.Textbox(label="LoRA")
|
| 613 |
with gr.Row():
|
|
|
|
| 567 |
css = ".container { max-width: 900px; margin: auto; }"
|
| 568 |
|
| 569 |
with gr.Blocks() as demo:
|
| 570 |
+
gr.Markdown("# 🧰SOONmerge® LoRA Toolkit")
|
| 571 |
|
| 572 |
with gr.Tabs():
|
| 573 |
+
with gr.Tab("Merge to Base + Reshard Output"):
|
| 574 |
t1_token = gr.Textbox(label="Token", type="password")
|
| 575 |
t1_base = gr.Textbox(label="Base Repo", value="ostris/Z-Image-De-Turbo")
|
| 576 |
t1_sub = gr.Textbox(label="Subfolder", value="transformer")
|
|
|
|
| 586 |
t1_res = gr.Textbox(label="Result")
|
| 587 |
t1_btn.click(task_merge, [t1_token, t1_base, t1_sub, t1_lora, t1_scale, t1_prec, t1_shard, t1_out, t1_struct, t1_priv], t1_res)
|
| 588 |
|
| 589 |
+
with gr.Tab("Extract Adapter"):
|
| 590 |
t2_token = gr.Textbox(label="Token", type="password")
|
| 591 |
t2_org = gr.Textbox(label="Original")
|
| 592 |
t2_tun = gr.Textbox(label="Tuned")
|
|
|
|
| 596 |
t2_res = gr.Textbox(label="Result")
|
| 597 |
t2_btn.click(task_extract, [t2_token, t2_org, t2_tun, t2_rank, t2_out], t2_res)
|
| 598 |
|
| 599 |
+
with gr.Tab("Merge Multiple Adapters"):
|
| 600 |
t3_token = gr.Textbox(label="Token", type="password")
|
| 601 |
t3_urls = gr.Textbox(label="URLs")
|
| 602 |
with gr.Row():
|
|
|
|
| 607 |
t3_res = gr.Textbox(label="Result")
|
| 608 |
t3_btn.click(task_merge_adapters, [t3_token, t3_urls, t3_beta, t3_sigma, t3_out], t3_res)
|
| 609 |
|
| 610 |
+
with gr.Tab("Resize Adapter"):
|
| 611 |
t4_token = gr.Textbox(label="Token", type="password")
|
| 612 |
t4_in = gr.Textbox(label="LoRA")
|
| 613 |
with gr.Row():
|