From c02ae4e9a22cfa2869c22822d1e3ee405f23b4e9 Mon Sep 17 00:00:00 2001 From: Mylo <36931363+gitmylo@users.noreply.github.com> Date: Sat, 19 Apr 2025 10:35:07 +0200 Subject: [PATCH] Update loader.py Co-authored-by: Akhilesh H --- flux_mod/loader.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/flux_mod/loader.py b/flux_mod/loader.py index 73960b1..f12813a 100644 --- a/flux_mod/loader.py +++ b/flux_mod/loader.py @@ -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")