diff --git a/README.md b/README.md index 4aabb55..ccbc452 100644 --- a/README.md +++ b/README.md @@ -8,7 +8,7 @@ A ComfyUI extension that tracks model usage frequency, helping you identify aban ## Features -- **Automatic Tracking** - Monitors usage of all model types: checkpoints, LoRAs, VAEs, ControlNets, CLIP, UNETs, upscalers, and more +- **Automatic Tracking** - Monitors usage of all model types: checkpoints, LoRAs, VAEs, ControlNets, CLIP, UNETs, upscalers, GGUF, and more - **Sidebar Panel** - View usage statistics directly in ComfyUI's sidebar - **Time Filtering** - View usage for all time, this month, or this week - **Smart Sorting** - Sort by last used, usage count, or name @@ -52,6 +52,7 @@ Restart ComfyUI after installation. | Upscalers | UpscaleModelLoader | | Style Models | StyleModelLoader | | GLIGEN | GLIGENLoader | +| GGUF | UnetLoaderGGUF, CLIPLoaderGGUF, DualCLIPLoaderGGUF, TripleCLIPLoaderGGUF, QuadrupleCLIPLoaderGGUF | Custom node loaders (Impact Pack, Efficiency Nodes, etc.) are also detected via pattern matching. diff --git a/js/modelpulse.js b/js/modelpulse.js index 4edca42..b0265cf 100644 --- a/js/modelpulse.js +++ b/js/modelpulse.js @@ -100,6 +100,7 @@ function getCategoryInfo(category) { upscaler: { name: "Upscalers", icon: "🔍" }, style_model: { name: "Style Models", icon: "🎨" }, gligen: { name: "GLIGEN", icon: "📐" }, + gguf: { name: "GGUF", icon: "🗜️" }, }; return categories[category] || { name: category, icon: "📁" }; } diff --git a/py/model_types.py b/py/model_types.py index cd105a3..097451a 100644 --- a/py/model_types.py +++ b/py/model_types.py @@ -36,6 +36,14 @@ MODEL_LOADERS: dict[str, tuple[str, str | list[str]]] = { # GLIGEN loaders "GLIGENLoader": ("gligen", "gligen_name"), + + # GGUF loaders (ComfyUI-GGUF by city96) + "UnetLoaderGGUF": ("gguf", "unet_name"), + "UnetLoaderGGUFAdvanced": ("gguf", "unet_name"), + "CLIPLoaderGGUF": ("gguf", "clip_name"), + "DualCLIPLoaderGGUF": ("gguf", ["clip_name1", "clip_name2"]), + "TripleCLIPLoaderGGUF": ("gguf", ["clip_name1", "clip_name2", "clip_name3"]), + "QuadrupleCLIPLoaderGGUF": ("gguf", ["clip_name1", "clip_name2", "clip_name3", "clip_name4"]), } # Patterns for detecting custom node loaders (Impact Pack, Efficiency Nodes, etc.) @@ -89,4 +97,8 @@ MODEL_CATEGORIES: dict[str, dict[str, str]] = { "name": "GLIGEN", "icon": "pi-th-large", }, + "gguf": { + "name": "GGUF", + "icon": "pi-database", + }, }