From 99f47084757ca4e1897e1d82deb3fa3d8e1f1629 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Kr=C3=BCger?= <valknar@pivoine.art>
Date: Sat, 22 Nov 2025 17:54:55 +0100
Subject: [PATCH] fix: correct ComfyUI directory types to match actual
 structure
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Changed type fields to use actual ComfyUI directory names:
- diffusion_models → checkpoints (for FLUX, CogVideoX)
- musicgen → checkpoints (with notes about custom_nodes placement)

All models now properly map to real ComfyUI/models/* subdirectories.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 comfyui_models.yaml | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/comfyui_models.yaml b/comfyui_models.yaml
index d914f70..c82c0f6 100644
--- a/comfyui_models.yaml
+++ b/comfyui_models.yaml
@@ -32,7 +32,7 @@ model_categories:
       size_gb: 23
       essential: true
       category: image
-      type: diffusion_models
+      type: checkpoints
       format: fp16
       vram_gb: 23
       notes: Industry-leading image generation quality
@@ -45,7 +45,7 @@ model_categories:
       size_gb: 23
       essential: false
       category: image
-      type: diffusion_models
+      type: checkpoints
       format: fp16
       vram_gb: 23
       notes: Development version with enhanced features
@@ -101,7 +101,7 @@ model_categories:
       size_gb: 20
       essential: true
       category: video
-      type: diffusion_models
+      type: checkpoints
       format: fp16
       vram_gb: 20
       frames: 49
@@ -150,11 +150,11 @@ model_categories:
       size_gb: 3
       essential: false
       category: audio
-      type: musicgen
+      type: checkpoints
       format: fp32
       vram_gb: 4
       duration_seconds: 30
-      notes: Fastest music generation, lower quality
+      notes: Fastest music generation, lower quality - Place in custom_nodes/ComfyUI_audio folder
       files:
         - source: "pytorch_model.bin"
           dest: "musicgen-small-pytorch_model.bin"
@@ -164,11 +164,11 @@ model_categories:
       size_gb: 11
       essential: true
       category: audio
-      type: musicgen
+      type: checkpoints
       format: fp32
       vram_gb: 8
       duration_seconds: 30
-      notes: Best balance of speed and quality
+      notes: Best balance of speed and quality - Place in custom_nodes/ComfyUI_audio folder
       files:
         - source: "pytorch_model.bin"
           dest: "musicgen-medium-pytorch_model.bin"
@@ -178,11 +178,11 @@ model_categories:
       size_gb: 22
       essential: false
       category: audio
-      type: musicgen
+      type: checkpoints
       format: fp32
       vram_gb: 16
       duration_seconds: 30
-      notes: Best quality, slower generation
+      notes: Best quality, slower generation - Place in custom_nodes/ComfyUI_audio folder
       files:
         - source: "pytorch_model-00001-of-00002.bin"
           dest: "musicgen-large-pytorch_model-00001-of-00002.bin"