From c85d036e616da468e7037cfa3f6f434bf839e713 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Kr=C3=BCger?= <valknar@pivoine.art>
Date: Sat, 22 Nov 2025 18:00:15 +0100
Subject: [PATCH] fix: restore correct model types after accidental changes
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- FLUX models: checkpoints → diffusion_models
- CogVideoX: checkpoints → diffusion_models
- MusicGen (all): checkpoints → musicgen
- Removed custom_nodes notes from MusicGen (using musicgen directory)

🤖 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 c82c0f6..d914f70 100644
--- a/comfyui_models.yaml
+++ b/comfyui_models.yaml
@@ -32,7 +32,7 @@ model_categories:
       size_gb: 23
       essential: true
       category: image
-      type: checkpoints
+      type: diffusion_models
       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: checkpoints
+      type: diffusion_models
       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: checkpoints
+      type: diffusion_models
       format: fp16
       vram_gb: 20
       frames: 49
@@ -150,11 +150,11 @@ model_categories:
       size_gb: 3
       essential: false
       category: audio
-      type: checkpoints
+      type: musicgen
       format: fp32
       vram_gb: 4
       duration_seconds: 30
-      notes: Fastest music generation, lower quality - Place in custom_nodes/ComfyUI_audio folder
+      notes: Fastest music generation, lower quality
       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: checkpoints
+      type: musicgen
       format: fp32
       vram_gb: 8
       duration_seconds: 30
-      notes: Best balance of speed and quality - Place in custom_nodes/ComfyUI_audio folder
+      notes: Best balance of speed and quality
       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: checkpoints
+      type: musicgen
       format: fp32
       vram_gb: 16
       duration_seconds: 30
-      notes: Best quality, slower generation - Place in custom_nodes/ComfyUI_audio folder
+      notes: Best quality, slower generation
       files:
         - source: "pytorch_model-00001-of-00002.bin"
           dest: "musicgen-large-pytorch_model-00001-of-00002.bin"