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"