Update checkpoint to FP8 e4m3fn, force 4 steps, disable cfg

app.py

@@ -105,12 +105,11 @@ def preload_model_files():
         local_dir = snapshot_download(
             repo_id="TSXu/Unicalli_Pro",
             allow_patterns=[
-                "*.safetensors",
-                "*.safetensors.index.json",
+                "unicalli_pro_chars7_dmd2_K4_fp8_e4m3fn.bin",
                 "internvl_embedding/*",  # All embedding files including tokenizer
                 "*.ttf",
             ],
-            ignore_patterns=["*.bin"],
+            ignore_patterns=["*.safetensors", "*.safetensors.index.json"],
             token=hf_token
         )
         print(f"✓ Unicalli_Pro cached at: {local_dir}")
@@ -304,6 +303,7 @@ def run_generation(text, font, author, num_steps, start_seed, num_images):
         result_img, cond_img = gen.generate(
             text=text, font_style=font, author=author,
             num_steps=num_steps, seed=current_seed,
+            guidance=1.0,
         )
         results.append((result_img, f"Seed: {current_seed}"))
         seeds_used.append(current_seed)
@@ -342,6 +342,9 @@ def interactive_session(
     yield "⏳ 加载模型并生成中... (首次需要1-2分钟编译) / Loading & generating... (first run ~1-2 min)", []
     progress(0.2, desc="处理中...")
     
+    # Hardcode num_steps to 4 for DMD distillation
+    num_steps = 4
+    
     results, seeds_used = run_generation(
         text, font, author, num_steps, start_seed, num_images
     )
@@ -410,12 +413,13 @@ with gr.Blocks(title="UniCalli - Chinese Calligraphy Generator / 中国书法生
             gr.Markdown("### ⚙️ 生成设置 / Generation Settings")
             
             num_steps = gr.Slider(
-                label="生成步数 / Inference Steps",
-                minimum=10,
-                maximum=50,
-                value=20,
+                label="生成步数 / Inference Steps (Fixed to 4 for DMD)",
+                minimum=4,
+                maximum=4,
+                value=4,
                 step=1,
-                info="更多步数 = 更高质量，但更慢 / More steps = higher quality, but slower"
+                interactive=False,
+                info="使用 DMD 蒸馏模型，强制 4 步生成 / Uses DMD distilled model, forced to 4 steps"
             )
             
             start_seed = gr.Number(