|
20 | 20 | from nodes import NODE_CLASS_MAPPINGS as GLOBAL_NODE_CLASS_MAPPINGS |
21 | 21 | from .nodes import ( |
22 | 22 | UnetLoaderGGUF, UnetLoaderGGUFAdvanced, |
23 | | - CLIPLoaderGGUF, DualCLIPLoaderGGUF, TripleCLIPLoaderGGUF, |
| 23 | + CLIPLoaderGGUF, DualCLIPLoaderGGUF, TripleCLIPLoaderGGUF, QuadrupleCLIPLoaderGGUF, |
24 | 24 | LTXVLoader, |
25 | 25 | Florence2ModelLoader, DownloadAndLoadFlorence2Model, |
26 | 26 | CheckpointLoaderNF4, |
@@ -693,6 +693,7 @@ def check_module_exists(module_path): |
693 | 693 | NODE_CLASS_MAPPINGS["CLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["CLIPLoader"]) |
694 | 694 | NODE_CLASS_MAPPINGS["DualCLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["DualCLIPLoader"]) |
695 | 695 | NODE_CLASS_MAPPINGS["TripleCLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["TripleCLIPLoader"]) |
| 696 | +NODE_CLASS_MAPPINGS["QuadrupleCLIPLoaderMultiGPU"] = override_class_clip(GLOBAL_NODE_CLASS_MAPPINGS["QuadrupleCLIPLoader"]) |
696 | 697 | NODE_CLASS_MAPPINGS["CheckpointLoaderSimpleMultiGPU"] = override_class(GLOBAL_NODE_CLASS_MAPPINGS["CheckpointLoaderSimple"]) |
697 | 698 | NODE_CLASS_MAPPINGS["ControlNetLoaderMultiGPU"] = override_class(GLOBAL_NODE_CLASS_MAPPINGS["ControlNetLoader"]) |
698 | 699 |
|
@@ -725,6 +726,8 @@ def check_module_exists(module_path): |
725 | 726 | NODE_CLASS_MAPPINGS["DualCLIPLoaderGGUFDisTorchMultiGPU"] = override_class_with_distorch_clip(DualCLIPLoaderGGUF) |
726 | 727 | NODE_CLASS_MAPPINGS["TripleCLIPLoaderGGUFMultiGPU"] = override_class_clip(TripleCLIPLoaderGGUF) |
727 | 728 | NODE_CLASS_MAPPINGS["TripleCLIPLoaderGGUFDisTorchMultiGPU"] = override_class_with_distorch_clip(TripleCLIPLoaderGGUF) |
| 729 | + NODE_CLASS_MAPPINGS["QuadrupleCLIPLoaderGGUFMultiGPU"] = override_class_clip(QuadrupleCLIPLoaderGGUF) |
| 730 | + NODE_CLASS_MAPPINGS["QuadrupleCLIPLoaderGGUFDisTorchMultiGPU"] = override_class_with_distorch_clip(QuadrupleCLIPLoaderGGUF) |
728 | 731 |
|
729 | 732 | if check_module_exists("PuLID_ComfyUI") or check_module_exists("pulid_comfyui"): |
730 | 733 | NODE_CLASS_MAPPINGS["PulidModelLoaderMultiGPU"] = override_class(PulidModelLoader) |
|
0 commit comments