微调UI

app.py

@@ -370,12 +370,17 @@ def warmup_model(pipeline: ZImagePipeline, resolutions: List[str]) -> None:
                 prompt=dummy_prompt,
                 resolution=res_str,
                 seed=42,
-                steps=6,
+                steps=9,
                 shift=3.0,
                 guidance_scale=0.0,
                 max_sequence_length=512,
                 use_lora=False,
                 lora_scale=0.0,
+                scheduler_name="FlowMatch Euler",
+                num_train_timesteps=1000,
+                use_dynamic_shifting=False,
+                base_shift=0.5,
+                max_shift=3.0,
             )
         except Exception as exc:  # noqa: BLE001
             print(f"Warmup failed for {res_str}: {exc}")
@@ -404,7 +409,7 @@ def generate(
     prompt: str,
     resolution: str = "1024x1024 ( 1:1 )",
     seed: int = 42,
-    steps: int = 9,
+    steps: int = 8,
     shift: float = 3.0,
     cfg: float = DEFAULT_CFG,
     scheduler_name: str = "FlowMatch Euler",
@@ -430,7 +435,7 @@ def generate(
             prompt=prompt,
             resolution=resolution.split(" ")[0] if " " in resolution else resolution,
             seed=new_seed,
-            steps=int(steps),
+            steps=int(steps) + 1,
             shift=float(shift),
             guidance_scale=float(cfg),
             max_sequence_length=int(max_sequence_length),
@@ -472,7 +477,7 @@ Model: `{MODEL_PATH}` | {pipe_status}
 
     with gr.Row():
         with gr.Column(scale=1):
-            prompt_input = gr.Textbox(label="Prompt", lines=3, placeholder="请输入提示词")
+            prompt_input = gr.Textbox(label="Prompt", lines=3, placeholder="Enter your prompt here...")
 
             with gr.Row():
                 choices = [int(k) for k in RES_CHOICES.keys()]
@@ -489,7 +494,7 @@ Model: `{MODEL_PATH}` | {pipe_status}
 
             with gr.Accordion("KSampler / Advanced", open=False):
                 with gr.Row():
-                    steps = gr.Slider(label="Steps", minimum=1, maximum=100, value=9, step=1)
+                    steps = gr.Slider(label="Steps", minimum=1, maximum=100, value=8, step=1)
                     cfg = gr.Slider(label="CFG", minimum=0.0, maximum=10.0, value=DEFAULT_CFG, step=0.1)
 
                 with gr.Row():
@@ -507,7 +512,7 @@ Model: `{MODEL_PATH}` | {pipe_status}
                     )
 
                 with gr.Row():
-                    shift = gr.Slider(label="Shift", minimum=0.0, maximum=10.0, value=3.0, step=0.1)
+                    shift = gr.Slider(label="Time Shift", minimum=1.0, maximum=10.0, value=3.0, step=0.1)
                     use_dynamic_shifting = gr.Checkbox(label="use_dynamic_shifting", value=False)
 
                 with gr.Row():
@@ -517,19 +522,19 @@ Model: `{MODEL_PATH}` | {pipe_status}
                 max_seq = gr.Slider(label="Max Sequence Length", minimum=256, maximum=1024, value=512, step=16)
 
             with gr.Row():
-                use_lora = gr.Checkbox(label="Use LoRA", value=True, interactive=lora_loaded)
+                use_lora = gr.Checkbox(label="Use LoRA", value=True, interactive=True)
                 lora_strength = gr.Slider(
                     label="LoRA Strength",
                     minimum=0.0,
                     maximum=1.5,
                     value=1.0,
                     step=0.05,
-                    interactive=lora_loaded,
+                    interactive=True,
                 )
 
             generate_btn = gr.Button("Generate", variant="primary")
 
-            gr.Markdown("### 示例提示词")
+            gr.Markdown("### 📝 Example Prompts")
             gr.Examples(examples=EXAMPLE_PROMPTS, inputs=prompt_input, label=None)
 
         with gr.Column(scale=1):