Add files via upload

2026-01-24 14:20:09 -05:00
parent 90e06565a3
commit 26154ccf66
2 changed files with 1047 additions and 0 deletions
@@ -0,0 +1,41 @@
+"""
+Z-Image-Turbo ComfyUI Nodes - Optimized & Verified
+===================================================
+Based on official Tongyi-MAI/Z-Image repository and HuggingFace diffusers implementation.
+
+CRITICAL FACTS about Z-Image-Turbo:
+1. Distilled using Decoupled-DMD on LINEAR sigma schedule (1.0 → 0.0)
+2. Optimal: 8-9 steps, CFG=1.0, Euler sampler
+3. Any deviation from linear schedule BREAKS the distillation
+4. The model expects sigma values as-is - no "recalibration" needed
+
+WHY CUSTOM SAMPLERS?
+ComfyUI's built-in euler uses the Karras ODE formulation with s_churn machinery.
+Our samplers use the clean interpolation formula which is:
+- More numerically stable (no division by small sigma)
+- Cleaner (no s_churn/s_tmin/s_tmax)
+- Explicit handling of final step (sigma_next = 0)
+
+Available samplers (in KSampler dropdown):
+- euler_flow: Clean 1st order Euler
+- euler_flow_alt: Derivative formulation
+- euler_flow_verbose: With debug output
+- midpoint_flow: 2nd order midpoint
+- heun_flow: 2nd order Heun
+
+Author: Capitan01R
+Version: 1.0.0
+"""
+
+from .zimage_turbo_core import (
+    NODE_CLASS_MAPPINGS,
+    NODE_DISPLAY_NAME_MAPPINGS,
+)
+
+__all__ = [
+    "NODE_CLASS_MAPPINGS",
+    "NODE_DISPLAY_NAME_MAPPINGS",
+]
+
+__version__ = "1.0.0"
+__author__ = "Capitan01R"