Test for gguf fix

2025-04-19 11:25:27 +02:00
1 changed files with 5 additions and 1 deletions
@@ -22,10 +22,14 @@ def ensure_gguf():
    if gguf:
        return
    import sys
-    gguf = sys.modules.get("ComfyUI-GGUF")
+    gguf = next(
+        (mod for path, mod in sys.modules.items() if path.endswith("ComfyUI-GGUF")),
+        None
+    )
    if gguf is None:
        raise RuntimeError("Could not find ComfyUI-GGUF node: GGUF support requires ComfyUI-GGUF")

+
 class ExternalFlux(comfy.supported_models_base.BASE):
    unet_config = {}
    unet_extra_config = {}