Update app.py

app.py

@@ -572,15 +572,15 @@ with gr.Blocks() as demo:
     with gr.Tabs():
         with gr.Tab("Merge to Base + Reshard Output"):
             t1_token = gr.Textbox(label="Token", type="password")
-            t1_base = gr.Textbox(label="Base Repo", value="ostris/Z-Image-De-Turbo")
+            t1_base = gr.Textbox(label="Base Repo (Diffusers)", value="ostris/Z-Image-De-Turbo")
             t1_sub = gr.Textbox(label="Subfolder", value="transformer")
-            t1_lora = gr.Textbox(label="LoRA")
+            t1_lora = gr.Textbox(label="LoRA Direct Link", value="https://huggingface.co/GuangyuanSD/Z-Image-Re-Turbo-LoRA/resolve/main/Z-image_re_turbo_lora_8steps_rank_32_v1_fp16.safetensors")
             with gr.Row():
                 t1_scale = gr.Slider(label="Scale", value=1.0, minimum=0, maximum=3.0, step=0.1)
                 t1_prec = gr.Radio(["bf16", "fp16", "float32"], value="bf16", label="Precision")
                 t1_shard = gr.Slider(label="Shard Size (GB)", value=2.0, minimum=0.1, maximum=10.0, step=0.1)
-            t1_out = gr.Textbox(label="Output")
-            t1_struct = gr.Textbox(label="Diffusers Structure Repo (Copies Base VAE/TextEnc/etc)", value="Tongyi-MAI/Z-Image-Turbo")
+            t1_out = gr.Textbox(label="Output Repo")
+            t1_struct = gr.Textbox(label="Diffusers Extras (Copies VAE/TextEnc/etc)", value="Tongyi-MAI/Z-Image-Turbo")
             t1_priv = gr.Checkbox(label="Private", value=True)
             t1_btn = gr.Button("Merge")
             t1_res = gr.Textbox(label="Result")
@@ -588,10 +588,10 @@ with gr.Blocks() as demo:
 
         with gr.Tab("Extract Adapter"):
             t2_token = gr.Textbox(label="Token", type="password")
-            t2_org = gr.Textbox(label="Original")
-            t2_tun = gr.Textbox(label="Tuned")
-            t2_rank = gr.Number(label="Rank", value=32, minimum=1, maximum=1024, step=1)
-            t2_out = gr.Textbox(label="Output")
+            t2_org = gr.Textbox(label="Original Model")
+            t2_tun = gr.Textbox(label="Tuned Model")
+            t2_rank = gr.Number(label="Extract At Rank", value=32, minimum=1, maximum=1024, step=1)
+            t2_out = gr.Textbox(label="Output Repo")
             t2_btn = gr.Button("Extract")
             t2_res = gr.Textbox(label="Result")
             t2_btn.click(task_extract, [t2_token, t2_org, t2_tun, t2_rank, t2_out], t2_res)
@@ -601,8 +601,8 @@ with gr.Blocks() as demo:
             t3_urls = gr.Textbox(label="URLs")
             with gr.Row():
                 t3_beta = gr.Slider(label="Beta", value=0.95, minimum=0.01, maximum=1.00, step=0.01)
-                t3_sigma = gr.Slider(label="Sigma Rel (Overrides Beta)", value=0.02, minimum=0.01, maximum=1.00, step=0.01)
-            t3_out = gr.Textbox(label="Output")
+                t3_sigma = gr.Slider(label="Sigma Rel (Overrides Beta)", value=0.21, minimum=0.01, maximum=1.00, step=0.01)
+            t3_out = gr.Textbox(label="Output Repo")
             t3_btn = gr.Button("Merge")
             t3_res = gr.Textbox(label="Result")
             t3_btn.click(task_merge_adapters, [t3_token, t3_urls, t3_beta, t3_sigma, t3_out], t3_res)
@@ -611,7 +611,7 @@ with gr.Blocks() as demo:
             t4_token = gr.Textbox(label="Token", type="password")
             t4_in = gr.Textbox(label="LoRA")
             with gr.Row():
-                t4_rank = gr.Number(label="Rank", value=8, minimum=1, maximum=256, step=1)
+                t4_rank = gr.Number(label="To Rank (Lower Only!)", value=8, minimum=1, maximum=256, step=1)
                 t4_method = gr.Dropdown(["None", "sv_ratio"], value="None", label="Dynamic Method")
                 t4_param = gr.Number(label="Dynamic Param", value=4.0)
             t4_out = gr.Textbox(label="Output")