From 2c6fb487d3a6dca9bd227e497a3e38ebc2ac6bde Mon Sep 17 00:00:00 2001
From: John Pollock <pollockjj@gmail.com>
Date: Sat, 28 Dec 2024 14:49:02 -0600
Subject: [PATCH] editing for clarity

---
 __init__.py | 55 +++++++----------------------------------------------
 1 file changed, 7 insertions(+), 48 deletions(-)

diff --git a/__init__.py b/__init__.py
index 072f070..b657035 100644
--- a/__init__.py
+++ b/__init__.py
@@ -85,54 +85,13 @@ def register_module(module_path, target_nodes):
 
 NODE_CLASS_MAPPINGS = {}
 
-# Let's test just one new module at a time, starting with GGUF
-logging.info("MultiGPU: Starting Core ComfyUI registration")
-register_module("", [
-    "UNETLoader",
-    "VAELoader", 
-    "CLIPLoader",
-    "DualCLIPLoader", 
-    "TripleCLIPLoader", 
-    "CheckpointLoaderSimple", 
-    "ControlNetLoader"
-])
-
-logging.info("MultiGPU: Starting GGUF registration")
-register_module("ComfyUI-GGUF", [
-    "UnetLoaderGGUF",
-    "UnetLoaderGGUFAdvanced",
-    "CLIPLoaderGGUF",
-    "DualCLIPLoaderGGUF",
-    "TripleCLIPLoaderGGUF"
-])
-logging.info("MultiGPU: Starting X-Flux ControlNet registration")
-register_module("x-flux-comfyui", [
-    "LoadFluxControlNet"
-])
-
-logging.info("MultiGPU: Starting Florence2 registration")
-register_module("ComfyUI-Florence2", [
-    "Florence2ModelLoader",
-    "DownloadAndLoadFlorence2Model"
-])
-
-logging.info("MultiGPU: Starting LTXVideo registration")
-register_module("ComfyUI-LTXVideo", [
-    "LTXVLoader"
-])
-
-logging.info("MultiGPU: Starting MMAudio registration")
-register_module("ComfyUI-MMAudio", [
-    "MMAudioFeatureUtilsLoader",
-    "MMAudioModelLoader",
-    "MMAudioSampler"
-])
-
-logging.info("MultiGPU: Starting NF4 registration")
-register_module("ComfyUI_bnb_nf4_fp4_Loaders", [
-    "CheckpointLoaderNF4",
-    "UNETLoaderNF4"
-])
+register_module("", ["UNETLoader", "VAELoader", "CLIPLoader", "DualCLIPLoader", "TripleCLIPLoader", "CheckpointLoaderSimple", "ControlNetLoader"])
+register_module("ComfyUI-GGUF", ["UnetLoaderGGUF","UnetLoaderGGUFAdvanced","CLIPLoaderGGUF","DualCLIPLoaderGGUF","TripleCLIPLoaderGGUF"])
+register_module("x-flux-comfyui", ["LoadFluxControlNet"])
+register_module("ComfyUI-Florence2", ["Florence2ModelLoader","DownloadAndLoadFlorence2Model"])
+register_module("ComfyUI-LTXVideo", ["LTXVLoader"])
+register_module("ComfyUI-MMAudio", ["MMAudioFeatureUtilsLoader","MMAudioModelLoader","MMAudioSampler"])
+register_module("ComfyUI_bitsandbytes_NF4", ["CheckpointLoaderNF4",])
 
 
 logging.info(f"MultiGPU: Registration complete. Final mappings: {', '.join(NODE_CLASS_MAPPINGS.keys())}")
\ No newline at end of file