Add file size display for models

- Show model file sizes (GB/MB) in the sidebar next to last used time
- Use ComfyUI's folder_paths for reliable model path resolution
- Support all configured model locations including extra_model_paths
- Handle GGUF models that can be in unet or clip folders
- Update README with new features
This commit is contained in:
Setmaster
2026-01-15 12:27:04 -05:00
parent daddb8d643
commit fb28135b82
4 changed files with 87 additions and 2 deletions
+14
View File
@@ -102,3 +102,17 @@ MODEL_CATEGORIES: dict[str, dict[str, str]] = {
"icon": "pi-database",
},
}
# Map categories to ComfyUI model folder names
CATEGORY_FOLDERS: dict[str, str] = {
"checkpoint": "checkpoints",
"lora": "loras",
"vae": "vae",
"controlnet": "controlnet",
"clip": "clip",
"unet": "unet",
"upscaler": "upscale_models",
"style_model": "style_models",
"gligen": "gligen",
"gguf": "unet", # GGUF models are stored in unet folder
}