chore: update .gitignore and add pre-commit configuration; refactor bridge.py and nodes.py for improved code readability and organization

Author: GACLove

.gitignore

@@ -306,4 +306,4 @@ pyrightconfig.json
 .ionide
 
 # End of https://www.toptal.com/developers/gitignore/api/python,visualstudiocode,pycharm
-.dev.md
+.dev.md

.pre-commit-config.yaml

@@ -0,0 +1,21 @@
+# Follow https://verdantfox.com/blog/how-to-use-git-pre-commit-hooks-the-hard-way-and-the-easy-way
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.11.0
+    hooks:
+      - id: ruff
+        args: [--fix, --respect-gitignore, --config=pyproject.toml]
+      - id: ruff-format
+        args: [--config=pyproject.toml]
+
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.5.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-yaml
+      - id: check-toml
+      - id: check-added-large-files
+      - id: check-case-conflict
+      - id: check-merge-conflict
+      - id: debug-statements

README_CN.md

@@ -171,4 +171,4 @@ ComfyUI/models/lightx2v/
 
 - `wan_i2v.json` - 基础图像生成视频
 - `wan_i2v_with_distill_lora.json` - 带蒸馏 LoRA 的 I2V
-- `wan_t2v_with_distill_lora.json` - 带蒸馏 LoRA 的 T2V
+- `wan_t2v_with_distill_lora.json` - 带蒸馏 LoRA 的 T2V

bridge.py

@@ -161,31 +161,79 @@ class CoefficientCalculator:
         "t2v": {
             "1.3b": {
                 "default": [
-                    [-5.21862437e04, 9.23041404e03, -5.28275948e02, 1.36987616e01, -4.99875664e-02],
-                    [2.39676752e03, -1.31110545e03, 2.01331979e02, -8.29855975e00, 1.37887774e-01],
+                    [
+                        -5.21862437e04,
+                        9.23041404e03,
+                        -5.28275948e02,
+                        1.36987616e01,
+                        -4.99875664e-02,
+                    ],
+                    [
+                        2.39676752e03,
+                        -1.31110545e03,
+                        2.01331979e02,
+                        -8.29855975e00,
+                        1.37887774e-01,
+                    ],
                 ]
             },
             "14b": {
                 "default": [
-                    [-3.03318725e05, 4.90537029e04, -2.65530556e03, 5.87365115e01, -3.15583525e-01],
-                    [-5784.54975374, 5449.50911966, -1811.16591783, 256.27178429, -13.02252404],
+                    [
+                        -3.03318725e05,
+                        4.90537029e04,
+                        -2.65530556e03,
+                        5.87365115e01,
+                        -3.15583525e-01,
+                    ],
+                    [
+                        -5784.54975374,
+                        5449.50911966,
+                        -1811.16591783,
+                        256.27178429,
+                        -13.02252404,
+                    ],
                 ]
             },
         },
         "i2v": {
             "720p": [
-                [8.10705460e03, 2.13393892e03, -3.72934672e02, 1.66203073e01, -4.17769401e-02],
+                [
+                    8.10705460e03,
+                    2.13393892e03,
+                    -3.72934672e02,
+                    1.66203073e01,
+                    -4.17769401e-02,
+                ],
                 [-114.36346466, 65.26524496, -18.82220707, 4.91518089, -0.23412683],
             ],
             "480p": [
-                [2.57151496e05, -3.54229917e04, 1.40286849e03, -1.35890334e01, 1.32517977e-01],
-                [-3.02331670e02, 2.23948934e02, -5.25463970e01, 5.87348440e00, -2.01973289e-01],
+                [
+                    2.57151496e05,
+                    -3.54229917e04,
+                    1.40286849e03,
+                    -1.35890334e01,
+                    1.32517977e-01,
+                ],
+                [
+                    -3.02331670e02,
+                    2.23948934e02,
+                    -5.25463970e01,
+                    5.87348440e00,
+                    -2.01973289e-01,
+                ],
             ],
         },
     }
 
     @classmethod
-    def get_coefficients(cls, task: str, model_size: str, resolution: Tuple[int, int], use_ret_steps: bool) -> List[List[float]]:
+    def get_coefficients(
+        cls,
+        task: str,
+        model_size: str,
+        resolution: Tuple[int, int],
+        use_ret_steps: bool,
+    ) -> List[List[float]]:
         """Get appropriate coefficients for TeaCache."""
         if task == "t2v":
             coeffs = cls.COEFFICIENTS["t2v"].get(model_size, {}).get("default", None)
@@ -269,13 +317,15 @@ def apply_inference_config(self, config: Dict[str, Any]) -> Dict[str, Any]:
             updates["target_video_length"] = config["video_length"]
         if "fps" in config:
             updates["fps"] = config["fps"]
-            
+
         if "denoising_step_list" in config:
             updates["denoising_step_list"] = config["denoising_step_list"]
 
         return updates
 
-    def apply_teacache_config(self, config: Dict[str, Any], model_info: Dict[str, Any]) -> Dict[str, Any]:
+    def apply_teacache_config(
+        self, config: Dict[str, Any], model_info: Dict[str, Any]
+    ) -> Dict[str, Any]:
         """Apply TeaCache configuration."""
         updates = {}
 
@@ -286,16 +336,23 @@ def apply_teacache_config(self, config: Dict[str, Any], model_info: Dict[str, An
 
             task = model_info.get("task", "t2v")
             model_size = "14b" if "14b" in model_info.get("model_cls", "") else "1.3b"
-            resolution = (model_info.get("target_width", 832), model_info.get("target_height", 480))
+            resolution = (
+                model_info.get("target_width", 832),
+                model_info.get("target_height", 480),
+            )
 
-            coeffs = CoefficientCalculator.get_coefficients(task, model_size, resolution, updates["use_ret_steps"])
+            coeffs = CoefficientCalculator.get_coefficients(
+                task, model_size, resolution, updates["use_ret_steps"]
+            )
             updates["coefficients"] = coeffs
         else:
             updates["feature_caching"] = "NoCaching"
 
         return updates
 
-    def apply_quantization_config(self, config: Dict[str, Any], model_path: str) -> Dict[str, Any]:
+    def apply_quantization_config(
+        self, config: Dict[str, Any], model_path: str
+    ) -> Dict[str, Any]:
         """Apply quantization configuration."""
         updates = {}
 
@@ -309,14 +366,18 @@ def apply_quantization_config(self, config: Dict[str, Any], model_path: str) ->
         updates["t5_quantized"] = t5_scheme != "bf16"
         if t5_scheme != "bf16":
             t5_path = os.path.join(model_path, t5_scheme)
-            updates["t5_quantized_ckpt"] = os.path.join(t5_path, f"models_t5_umt5-xxl-enc-{t5_scheme}.pth")
+            updates["t5_quantized_ckpt"] = os.path.join(
+                t5_path, f"models_t5_umt5-xxl-enc-{t5_scheme}.pth"
+            )
 
         clip_scheme = config.get("clip_precision", "fp16")
         updates["clip_quant_scheme"] = clip_scheme
         updates["clip_quantized"] = clip_scheme != "fp16"
         if clip_scheme != "fp16":
             clip_path = os.path.join(model_path, clip_scheme)
-            updates["clip_quantized_ckpt"] = os.path.join(clip_path, f"clip-{clip_scheme}.pth")
+            updates["clip_quantized_ckpt"] = os.path.join(
+                clip_path, f"clip-{clip_scheme}.pth"
+            )
 
         quant_backend = config.get("quant_backend", "vllm")
         updates["quant_op"] = quant_backend
@@ -330,7 +391,9 @@ def apply_quantization_config(self, config: Dict[str, Any], model_path: str) ->
                 else:
                     mm_type = f"W-{dit_scheme}-channel-sym-A-{dit_scheme}-channel-sym-dynamic-Sgl"
             elif quant_backend == "q8f":
-                mm_type = f"W-{dit_scheme}-channel-sym-A-{dit_scheme}-channel-sym-dynamic-Q8F"
+                mm_type = (
+                    f"W-{dit_scheme}-channel-sym-A-{dit_scheme}-channel-sym-dynamic-Q8F"
+                )
             else:
                 mm_type = "Default"
 
@@ -357,15 +420,21 @@ def apply_memory_optimization(self, config: Dict[str, Any]) -> Dict[str, Any]:
             updates["clean_cuda_cache"] = True
 
         # CPU offloading
-        if config.get("enable_cpu_offload", False) or level in ["medium", "high", "extreme"]:
+        if config.get("enable_cpu_offload", False) or level in [
+            "medium",
+            "high",
+            "extreme",
+        ]:
             updates["cpu_offload"] = True
             updates["offload_granularity"] = config.get("offload_granularity", "phase")
             updates["offload_ratio"] = config.get("offload_ratio", 1.0)
 
         # T5 offloading
         if level in ["high", "extreme"]:
             updates["t5_cpu_offload"] = True
-            updates["t5_offload_granularity"] = "block" if level == "extreme" else "model"
+            updates["t5_offload_granularity"] = (
+                "block" if level == "extreme" else "model"
+            )
 
         # Module management
         if config.get("lazy_load", False) or level == "extreme":
@@ -383,7 +452,9 @@ def apply_memory_optimization(self, config: Dict[str, Any]) -> Dict[str, Any]:
 
         return updates
 
-    def apply_vae_config(self, config: Dict[str, Any], model_path: str) -> Dict[str, Any]:
+    def apply_vae_config(
+        self, config: Dict[str, Any], model_path: str
+    ) -> Dict[str, Any]:
         """Apply VAE configuration."""
         updates = {}
 
@@ -413,7 +484,9 @@ def build_final_config(self, configs: Dict[str, Dict[str, Any]]) -> EasyDict:
 
         if "quantization" in configs:
             model_path = final_config.get("model_path", "")
-            quant_updates = self.apply_quantization_config(configs["quantization"], model_path)
+            quant_updates = self.apply_quantization_config(
+                configs["quantization"], model_path
+            )
             final_config.update(quant_updates)
 
         if "memory" in configs:

examples/wan_i2v.json

@@ -367,4 +367,4 @@
     "frontendVersion": "1.19.9"
   },
   "version": 0.4
-}
+}