Update loader.py

Co-authored-by: Akhilesh H <itsakza07@gmail.com>
2025-04-19 10:35:07 +02:00
parent 20bd50e0e3
commit c02ae4e9a2
1 changed files with 4 additions and 6 deletions
@@ -22,12 +22,10 @@ def ensure_gguf():
    if gguf:
        return
    import sys
-    gguf = sys.modules.get("ComfyUI-GGUF")
-
-    gguf.loader = sys.modules.get("ComfyUI-GGUF.loader")
-    gguf.ops = sys.modules.get("ComfyUI-GGUF.ops")
-    gguf.nodes = sys.modules.get("ComfyUI-GGUF.nodes")
-    
+    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")