diff --git a/src/platform/assets/utils/categoryLabel.ts b/src/platform/assets/utils/categoryLabel.ts
index 354249b2d..e5a655333 100644
--- a/src/platform/assets/utils/categoryLabel.ts
+++ b/src/platform/assets/utils/categoryLabel.ts
@@ -3,6 +3,9 @@ const ACRONYM_TAGS = new Set(['VAE', 'CLIP', 'GLIGEN'])
 export function formatCategoryLabel(raw?: string): string {
   if (!raw) return 'Models'
 
+  // Special display name mappings
+  if (raw === 'diffusion_models') return 'Diffusion'
+
   return raw
     .split('_')
     .map((segment) => {