diff --git a/comfyui_models.yaml b/comfyui_models.yaml
index 67feed6..64ddc70 100644
--- a/comfyui_models.yaml
+++ b/comfyui_models.yaml
@@ -32,7 +32,7 @@ model_categories:
       size_gb: 23
       essential: true
       category: image
-      type: diffusers
+      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: diffusers
+      type: diffusion_models
       format: fp16
       vram_gb: 23
       notes: Development version with enhanced features
@@ -58,7 +58,7 @@ model_categories:
       size_gb: 7
       essential: true
       category: image
-      type: diffusers
+      type: checkpoints
       format: fp16
       vram_gb: 12
       notes: Most widely used Stable Diffusion model
@@ -71,7 +71,7 @@ model_categories:
       size_gb: 6
       essential: false
       category: image
-      type: diffusers
+      type: checkpoints
       format: fp16
       vram_gb: 12
       notes: Use after SDXL base for improved details
@@ -84,7 +84,7 @@ model_categories:
       size_gb: 18
       essential: false
       category: image
-      type: diffusers
+      type: checkpoints
       format: fp16
       vram_gb: 20
       notes: Newest generation Stable Diffusion
@@ -116,7 +116,7 @@ model_categories:
       size_gb: 8
       essential: true
       category: video
-      type: diffusion_models
+      type: checkpoints
       format: fp16
       vram_gb: 20
       frames: 14
@@ -131,7 +131,7 @@ model_categories:
       size_gb: 8
       essential: false
       category: video
-      type: diffusion_models
+      type: checkpoints
       format: fp16
       vram_gb: 20
       frames: 25
@@ -150,7 +150,7 @@ model_categories:
       size_gb: 3
       essential: false
       category: audio
-      type: audio_models
+      type: musicgen
       format: fp32
       vram_gb: 4
       duration_seconds: 30
@@ -164,7 +164,7 @@ model_categories:
       size_gb: 11
       essential: true
       category: audio
-      type: audio_models
+      type: musicgen
       format: fp32
       vram_gb: 8
       duration_seconds: 30
@@ -178,7 +178,7 @@ model_categories:
       size_gb: 22
       essential: false
       category: audio
-      type: audio_models
+      type: musicgen
       format: fp32
       vram_gb: 16
       duration_seconds: 30