ResolutionSpeedupChecker (#1102)

![workflow
(3)](https://github.com/user-attachments/assets/6f5839f2-1139-4e0b-8518-ea559842d909)


<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->
## Summary by CodeRabbit

- **New Features**
- Introduced a `ResolutionSpeedupChecker` class for validating input
sample dimensions against a configuration.
- Implemented a mechanism to enable or disable deployable functionality
via a global variable and associated functions.

- **Bug Fixes**
- Enhanced the constructor of the `Hijacker` class to ensure robust
handling of the `funcs_list` parameter.

- **Documentation**
- Updated mappings to include the new `ResolutionSpeedupChecker` class
for better recognition within the system.
<!-- end of auto-generated comment: release notes by coderabbit.ai -->

Author: ccssu

onediff_comfy_nodes/extras_nodes/nodes_nexfort_booster.py

@@ -1,4 +1,8 @@
 import collections
+import os
+
+import yaml
+from onediff.infer_compiler.backends.nexfort import fallback_to_eager
 
 from ..modules.nexfort.booster_basic import BasicNexFortBoosterExecutor
 
@@ -52,8 +56,40 @@ def apply(
         )
 
 
+class ResolutionSpeedupChecker:
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    config_path = os.path.join(current_dir, "resolutions_config.yaml")
+    with open(config_path, "r") as file:
+        resolutions = yaml.safe_load(file)
+    height_width_dict = {x["height"]: x["width"] for x in resolutions["resolutions"]}
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "samples": ("LATENT",),
+            }
+        }
+
+    RETURN_TYPES = ("LATENT",)
+    FUNCTION = "check"
+
+    def check(self, samples):
+        _, _, H, W = samples["samples"].shape
+        H, W = H * 8, W * 8
+        if H in self.height_width_dict and self.height_width_dict[H] == W:
+            fallback_to_eager(True)
+        else:
+            fallback_to_eager(False)
+        return (samples,)
+
+
 NODE_CLASS_MAPPINGS = {
     "OneDiffNexfortBooster": OneDiffNexfortBooster,
+    "ResolutionSpeedupChecker": ResolutionSpeedupChecker,
 }
 
-NODE_DISPLAY_NAME_MAPPINGS = {"OneDiffNexfortBooster": "Nexfort Booster - OneDiff"}
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "OneDiffNexfortBooster": "Nexfort Booster - OneDiff",
+    "ResolutionSpeedupChecker": "Speedup Checker - Resolution",
+}

onediff_comfy_nodes/extras_nodes/resolutions_config.yaml

@@ -0,0 +1,31 @@
+resolutions:
+  - width: 1024
+    height: 1024
+    aspect_ratio: '1:1 Square'
+  - width: 1152
+    height: 896
+    aspect_ratio: '9:7'
+  - width: 896
+    height: 1152
+    aspect_ratio: '7:9'
+  - width: 1216
+    height: 832
+    aspect_ratio: '19:13'
+  - width: 832
+    height: 1216
+    aspect_ratio: '13:19'
+  - width: 1344
+    height: 768
+    aspect_ratio: '7:4 Horizontal'
+  - width: 768
+    height: 1344
+    aspect_ratio: '4:7 Vertical'
+  - width: 1536
+    height: 640
+    aspect_ratio: '12:5 Horizontal'
+  - width: 640
+    height: 1536
+    aspect_ratio: '5:12 Vertical'
+  - width: 512
+    height: 512
+    aspect_ratio: '1:1 Square'

onediff_comfy_nodes/modules/sd_hijack_utils.py

@@ -162,7 +162,10 @@ class Hijacker:
     """
 
     def __init__(self, funcs_list=[]):
-        self.funcs_list = funcs_list
+        if funcs_list and isinstance(funcs_list, List):
+            self.funcs_list = funcs_list.copy()
+        else:
+            self.funcs_list = []
         self.unhijack_funcs = []
 
     def hijack(self, last=True):

src/onediff/infer_compiler/backends/nexfort/__init__.py

@@ -1 +1,2 @@
 from . import nexfort as _nexfort_backend
+from .deployable_module import fallback_to_eager

src/onediff/infer_compiler/backends/nexfort/deployable_module.py

@@ -6,6 +6,19 @@
 
 from ..deployable_module import DeployableModule
 
+DISABLE_DEPLOYABLE = False
+
+
+def fallback_to_eager(value: bool = False):
+    """
+    Set the DISABLE_DEPLOYABLE environment variable.
+
+    Args:
+        value (bool): The value to set for the environment variable.
+    """
+    global DISABLE_DEPLOYABLE
+    DISABLE_DEPLOYABLE = not value
+
 
 class NexfortDeployableModule(DeployableModule):
     def __init__(self, compiled_module, torch_module):
@@ -23,6 +36,8 @@ def __init__(self, compiled_module, torch_module):
             object.__setattr__(self, "_buffers", compiled_module._orig_mod._buffers)
 
     def forward(self, *args, **kwargs):
+        if DISABLE_DEPLOYABLE:
+            return self._torch_module(*args, **kwargs)
         with torch._dynamo.utils.disable_cache_limit():
             return self._deployable_module_model(*args, **kwargs)
 
@@ -34,6 +49,8 @@ def _create_deployable_function(
     compiled_model, torch_module: FunctionType = None
 ) -> FunctionType:
     def deploy_function(*args, **kwargs):
+        if DISABLE_DEPLOYABLE:
+            return torch_module(*args, **kwargs)
         with torch._dynamo.utils.disable_cache_limit():
             return compiled_model(*args, **kwargs)