Skip to content

Commit 79e9230

Browse files
committed
fix(gguf): correct missing type enum for CLIPLoaderGGUF and DualCLIPLoaderGGUF
Populate 'type' options by sourcing from core nodes to avoid drift:\n- CLIPLoaderGGUF now derives 'type' from nodes.CLIPLoader.INPUT_TYPES()\n- DualCLIPLoaderGGUF now derives 'type' from nodes.DualCLIPLoader.INPUT_TYPES()\nThis fixes missing or outdated 'type' options in GGUF Single and Dual CLIP loaders.\n\nchore: bump version to 1.8.2
1 parent 7a08dd9 commit 79e9230

3 files changed

Lines changed: 15 additions & 5 deletions

File tree

__init__.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -590,11 +590,17 @@ def check_module_exists(module_path):
590590
NODE_CLASS_MAPPINGS["VAELoaderMultiGPU"] = override_class(GLOBAL_NODE_CLASS_MAPPINGS["VAELoader"])
591591
NODE_CLASS_MAPPINGS["CLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["CLIPLoader"])
592592
NODE_CLASS_MAPPINGS["DualCLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["DualCLIPLoader"])
593-
NODE_CLASS_MAPPINGS["TripleCLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["TripleCLIPLoader"])
594-
NODE_CLASS_MAPPINGS["QuadrupleCLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["QuadrupleCLIPLoader"])
593+
if "TripleCLIPLoader" in GLOBAL_NODE_CLASS_MAPPINGS:
594+
NODE_CLASS_MAPPINGS["TripleCLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["TripleCLIPLoader"])
595+
if "QuadrupleCLIPLoader" in GLOBAL_NODE_CLASS_MAPPINGS:
596+
NODE_CLASS_MAPPINGS["QuadrupleCLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["QuadrupleCLIPLoader"])
595597
NODE_CLASS_MAPPINGS["CLIPVisionLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["CLIPVisionLoader"])
596598
NODE_CLASS_MAPPINGS["CheckpointLoaderSimpleMultiGPU"] = override_class(GLOBAL_NODE_CLASS_MAPPINGS["CheckpointLoaderSimple"])
597599
NODE_CLASS_MAPPINGS["ControlNetLoaderMultiGPU"] = override_class(GLOBAL_NODE_CLASS_MAPPINGS["ControlNetLoader"])
600+
if "DiffusersLoader" in GLOBAL_NODE_CLASS_MAPPINGS:
601+
NODE_CLASS_MAPPINGS["DiffusersLoaderMultiGPU"] = override_class(GLOBAL_NODE_CLASS_MAPPINGS["DiffusersLoader"])
602+
if "DiffControlNetLoader" in GLOBAL_NODE_CLASS_MAPPINGS:
603+
NODE_CLASS_MAPPINGS["DiffControlNetLoaderMultiGPU"] = override_class(GLOBAL_NODE_CLASS_MAPPINGS["DiffControlNetLoader"])
598604

599605
if check_module_exists("ComfyUI-LTXVideo") or check_module_exists("comfyui-ltxvideo"):
600606
NODE_CLASS_MAPPINGS["LTXVLoaderMultiGPU"] = override_class(LTXVLoader)

nodes.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,10 +40,12 @@ def INPUT_TYPES(s):
4040
class CLIPLoaderGGUF:
4141
@classmethod
4242
def INPUT_TYPES(s):
43+
import nodes
44+
base = nodes.CLIPLoader.INPUT_TYPES()
4345
return {
4446
"required": {
4547
"clip_name": (s.get_filename_list(),),
46-
"type": (["stable_diffusion", "stable_cascade", "sd3", "stable_audio", "mochi", "ltxv", "pixart", "wan"],),
48+
"type": base["required"]["type"],
4749
}
4850
}
4951

@@ -77,12 +79,14 @@ def load_clip(self, clip_name, type="stable_diffusion"):
7779
class DualCLIPLoaderGGUF(CLIPLoaderGGUF):
7880
@classmethod
7981
def INPUT_TYPES(s):
82+
import nodes
83+
base = nodes.DualCLIPLoader.INPUT_TYPES()
8084
file_options = (s.get_filename_list(), )
8185
return {
8286
"required": {
8387
"clip_name1": file_options,
8488
"clip_name2": file_options,
85-
"type": (("sdxl", "sd3", "flux", "hunyuan_video"),),
89+
"type": base["required"]["type"],
8690
}
8791
}
8892

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
[project]
22
name = "comfyui-multigpu"
33
description = "Adds full multi-GPU support for WanVideoWrapper, enabling model loading and block-swapping on any device. Provides a suite of custom nodes to manage multiple GPUs for ComfyUI, including advanced GGUF offloading with DisTorch and device overrides for core nodes."
4-
version = "1.8.1"
4+
version = "1.8.2"
55
license = {file = "LICENSE"}
66

77
[project.urls]

0 commit comments

Comments
 (0)