feat(turbo): add native 4-step and 8-step LightX2V distill Turbo LoRA acceleration

app.py

@@ -1,4 +1,4 @@
-"""Wan 2.2 Space with ZeroGPU, Multi-LoRA (Civitai + HF), Presets, Trigger Words, and 300-Hours Optimizations."""
+"""Wan 2.2 Space with ZeroGPU, Turbo 4-8 Step Distill, Multi-LoRA (Civitai + HF), Presets, Trigger Words, and 300-Hours Optimizations."""
 
 from __future__ import annotations
 
@@ -33,6 +33,35 @@ LOCAL_LORAS_DIR = os.path.join(os.path.dirname(os.path.abspath(__file__)), "lora
 LORA_MAX_BYTES = 2 * 1024**3
 DEFAULT_FPS = 16
 
+# =========================================================================
+# Turbo Distillation LoRAs (4-Step & 8-Step Acceleration)
+# =========================================================================
+TURBO_PRESETS = {
+    "⚡ Turbo 4-Passos (LightX2V Distill - 4 Steps Ultra Rápido)": {
+        "wan2.2_i2v": ("lightx2v/Wan2.2-Distill-Loras", "wan2.2_i2v_A14b_high_noise_lora_rank64_lightx2v_4step_1022.safetensors"),
+        "wan2.2_t2v": ("lightx2v/Wan2.2-Distill-Loras", "wan2.2_t2v_A14b_high_noise_lora_rank64_lightx2v_4step_1217.safetensors"),
+        "wan2.1_i2v": ("lightx2v/Wan2.1-Distill-Loras", "wan2.1_i2v_lora_rank64_lightx2v_4step.safetensors"),
+        "wan2.1_t2v": ("lightx2v/Wan2.1-Distill-Loras", "wan2.1_t2v_14b_lora_rank64_lightx2v_4step.safetensors"),
+        "default_steps": 4,
+        "default_cfg": 1.5,
+        "description": "Destilação 4-passos LightX2V. Renderiza o vídeo em ~15-25 segundos mantendo alta qualidade!",
+    },
+    "⚡ Turbo 8-Passos (8 Steps Balanceado)": {
+        "wan2.2_i2v": ("lightx2v/Wan2.2-Distill-Loras", "wan2.2_i2v_A14b_low_noise_lora_rank64_lightx2v_4step_1022.safetensors"),
+        "wan2.2_t2v": ("lightx2v/Wan2.2-Distill-Loras", "wan2.2_t2v_A14b_low_noise_lora_rank64_lightx2v_4step_1217.safetensors"),
+        "wan2.1_i2v": ("lightx2v/Wan2.1-Distill-Loras", "wan2.1_i2v_lora_rank64_lightx2v_4step.safetensors"),
+        "wan2.1_t2v": ("lightx2v/Wan2.1-Distill-Loras", "wan2.1_t2v_14b_lora_rank64_lightx2v_4step.safetensors"),
+        "default_steps": 8,
+        "default_cfg": 2.0,
+        "description": "Destilação 8-passos para equilíbrio perfeito entre velocidade e detalhes.",
+    },
+    "🚫 Desativado (Modo Normal - 15 a 25 Passos)": {
+        "default_steps": 20,
+        "default_cfg": 5.0,
+        "description": "Modo padrão sem aceleração Turbo (requer 15-25 passos).",
+    },
+}
+
 # =========================================================================
 # Base NSFW Adapters (Unlocks adult generation on Wan Base Models)
 # =========================================================================
@@ -312,6 +341,31 @@ def _download_lora_url(url: str, civitai_token: str = "") -> tuple[str, str]:
     raise ValueError("Too many redirects downloading LoRA.")
 
 
+def resolve_turbo_lora(turbo_choice: str, model_repo: str) -> tuple[str | None, float]:
+    """Resolves and downloads the appropriate 4-step or 8-step distillation LoRA."""
+    if turbo_choice not in TURBO_PRESETS or "Desativado" in turbo_choice:
+        return None, 0.0
+
+    spec = TURBO_PRESETS[turbo_choice]
+    is_i2v = "I2V" in model_repo or "i2v" in model_repo or "TI2V" in model_repo
+    is_wan22 = "Wan2.2" in model_repo or "wan2.2" in model_repo
+
+    if is_wan22:
+        key = "wan2.2_i2v" if is_i2v else "wan2.2_t2v"
+    else:
+        key = "wan2.1_i2v" if is_i2v else "wan2.1_t2v"
+
+    if key not in spec:
+        return None, 0.0
+
+    repo_id, filename = spec[key]
+    from huggingface_hub import hf_hub_download
+    local_dir = _lora_cache_directory(f"hf://{repo_id}/{filename}")
+    path = hf_hub_download(repo_id=repo_id, filename=filename, token=False, local_dir=local_dir)
+    os.utime(local_dir, None)
+    return path, 1.0
+
+
 def resolve_single_lora(
     preset_type: str, custom_url: str, hf_repo: str, hf_file: str, local_file: str, strength: float, civitai_token: str = ""
 ) -> tuple[str | None, str, float]:
@@ -447,7 +501,7 @@ def _fit_keyframe(image_input, target_width: int, target_height: int) -> Image.I
 
 def get_duration(*args, **kwargs):
     try:
-        steps = kwargs.get("steps", args[6] if len(args) > 6 else 20)
+        steps = kwargs.get("steps", args[6] if len(args) > 6 else 4)
         return max(180, min(MAX_GPU_DURATION, int(steps) * 8 + 60))
     except Exception:
         return 240
@@ -529,6 +583,7 @@ def generate_video(
     prompt: str,
     negative_prompt: str,
     input_image: Image.Image | str | None,
+    turbo_choice: str,
     nsfw_base_choice: str,
     nsfw_base_strength: float,
     canvas: str,
@@ -571,6 +626,12 @@ def generate_video(
     lora_configs = []
     active_labels = []
 
+    # 0. Turbo LoRA (4-Step or 8-Step Distill)
+    t_path, t_scale = resolve_turbo_lora(turbo_choice, model_choice)
+    if t_path and t_scale > 0:
+        lora_configs.append((t_path, t_scale))
+        active_labels.append(f"⚡ Turbo: {turbo_choice.split('(')[0].strip()}")
+
     # 1. Base NSFW Adapter
     if nsfw_base_choice in NSFW_BASE_ADAPTERS and nsfw_base_strength > 0:
         base_spec = NSFW_BASE_ADAPTERS[nsfw_base_choice]
@@ -664,6 +725,11 @@ custom_css = """
     font-weight: 600;
     margin-right: 6px;
 }
+.badge-turbo {
+    background: rgba(245, 158, 11, 0.2);
+    color: #fbbf24;
+    border: 1px solid rgba(245, 158, 11, 0.4);
+}
 .badge-zerogpu {
     background: rgba(16, 185, 129, 0.2);
     color: #34d399;
@@ -711,17 +777,18 @@ all_preset_choices = (
     + (["Local File (loras/ folder)"] if get_local_loras() else [])
 )
 
-with gr.Blocks(css=custom_css, title="Wan 2.2 Video Studio") as app:
+with gr.Blocks(css=custom_css, title="Wan 2.2 Turbo Video Studio") as app:
     gr.HTML(
         """
         <div class="header-card">
             <div style="margin-bottom: 12px;">
+                <span class="badge badge-turbo">⚡ Turbo 4-Step Distill</span>
                 <span class="badge badge-zerogpu">⚡ ZeroGPU</span>
                 <span class="badge badge-model">🎬 Wan 2.2 (MoE Diffusers)</span>
                 <span class="badge badge-multilora">🧩 Base NSFW + Multi-LoRA Engine</span>
             </div>
-            <h1>Wan 2.2 AI Video Studio</h1>
-            <p>Image-to-Video (I2V) & Text-to-Video (T2V) com Arquitetura MoE Wan 2.2, Adaptador NSFW Base e Multi-LoRA.</p>
+            <h1>Wan 2.2 Turbo Video Studio</h1>
+            <p>Aceleração Turbo de 4 a 8 Passos com Arquitetura Wan 2.2 MoE, Adaptador Base NSFW e Multi-LoRA.</p>
         </div>
         """
     )
@@ -758,6 +825,14 @@ with gr.Blocks(css=custom_css, title="Wan 2.2 Video Studio") as app:
                 hmnsfw_example_btn = gr.Button("💡 Inserir Exemplo Completo de Prompt HMNSFW", elem_classes=["trigger-btn"])
 
             with gr.Tabs():
+                with gr.TabItem("⚡ Aceleração Turbo (4-8 Passos)"):
+                    gr.Markdown("Acelera a geração em até **8x** aplicando destilação LightX2V. Permite renderizar vídeos em ~15 segundos!")
+                    turbo_choice = gr.Dropdown(
+                        label="Modo Turbo",
+                        choices=list(TURBO_PRESETS.keys()),
+                        value="⚡ Turbo 4-Passos (LightX2V Distill - 4 Steps Ultra Rápido)",
+                    )
+
                 with gr.TabItem("🔞 Adaptador Base NSFW (Essencial)"):
                     gr.Markdown("Como o modelo base do Wan é treinado sem conteúdo explícito, este adaptador é carregado como base para desbloquear anatomia sem gastar seus slots de LoRA!")
                     with gr.Row():
@@ -883,11 +958,11 @@ with gr.Blocks(css=custom_css, title="Wan 2.2 Video Studio") as app:
                     )
                     steps = gr.Slider(
                         label="Passos de Inferência (Steps)",
-                        minimum=4,
+                        minimum=2,
                         maximum=40,
                         step=1,
-                        value=20,
-                        info="O guia recomenda 15-25 passos para qualidade ideal.",
+                        value=4,
+                        info="Com Turbo 4-Passos ativado, 4 passos fornecem render ultrarrápido.",
                     )
 
                 with gr.Row():
@@ -896,7 +971,8 @@ with gr.Blocks(css=custom_css, title="Wan 2.2 Video Studio") as app:
                         minimum=1.0,
                         maximum=10.0,
                         step=0.5,
-                        value=5.0,
+                        value=1.5,
+                        info="No modo Turbo, recomenda-se CFG baixo (1.0 a 2.0).",
                     )
                     seed = gr.Number(label="Seed", value=42, precision=0)
                     randomize_seed = gr.Checkbox(label="🎲 Randomizar Seed", value=True)
@@ -908,13 +984,23 @@ with gr.Blocks(css=custom_css, title="Wan 2.2 Video Studio") as app:
                     type="password",
                 )
 
-            generate_btn = gr.Button("🚀 Gerar Vídeo Wan 2.2", variant="primary", elem_classes=["generate-btn"])
+            generate_btn = gr.Button("🚀 Gerar Vídeo Wan 2.2 Turbo", variant="primary", elem_classes=["generate-btn"])
 
         with gr.Column(scale=6):
             output_video = gr.Video(label="Vídeo Gerado", autoplay=True, loop=True)
             output_report = gr.Markdown(label="Detalhes da Geração", value="Pronto para renderizar vídeo.")
 
     # Event handlers
+    def on_turbo_change(turbo_val):
+        spec = TURBO_PRESETS.get(turbo_val, {})
+        return spec.get("default_steps", 4), spec.get("default_cfg", 1.5)
+
+    turbo_choice.change(
+        fn=on_turbo_change,
+        inputs=[turbo_choice],
+        outputs=[steps, guidance_scale],
+    )
+
     def on_lora1_change(preset_val):
         spec = LORA_PRESETS.get(preset_val, {})
         triggers = spec.get("trigger_words", "")
@@ -991,6 +1077,7 @@ with gr.Blocks(css=custom_css, title="Wan 2.2 Video Studio") as app:
             prompt,
             negative_prompt,
             input_image,
+            turbo_choice,
             nsfw_base_choice,
             nsfw_base_strength,
             canvas,