xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-01-23 18:54:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

more robust GGUF node detection

Browse Source
This commit is contained in:
kijai 2025-09-19 02:05:04 +03:00
parent 3765985982
commit e81f33508b
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
nodes/model_optimization_nodes.py
View File

@ -1,3 +1,4 @@
import os
from comfy.ldm.modules import attention as comfy_attention
import logging
import comfy.model_patcher
@ -1963,15 +1964,20 @@ if v3_available:
@classmethod
def _get_gguf_module(cls):
"""Lazy import of GGUF module with fallback options"""
try:
return importlib.import_module("ComfyUI-GGUF")
except ImportError:
gguf_path = os.path.join(folder_paths.folder_names_and_paths["custom_nodes"][0][0], "ComfyUI-GGUF")
"""Import GGUF module with version validation"""
for module_name in ["ComfyUI-GGUF", "custom_nodes.ComfyUI-GGUF", "comfyui-gguf", "custom_nodes.comfyui-gguf", gguf_path, gguf_path.lower()]:
try:
return importlib.import_module("comfyui-gguf")
module = importlib.import_module(module_name)
return module
except ImportError:
raise ImportError("This node requires ComfyUI-GGUF to be installed.")
continue
raise ImportError(
"Compatible ComfyUI-GGUF not found. "
"Please install/update from: https://github.com/city96/ComfyUI-GGUF"
)
@classmethod
def execute(cls, model_name, extra_model_name, dequant_dtype, patch_dtype, patch_on_device, attention_override, enable_fp16_accumulation):