[refactor] Move model utility functions to model_utils module

2025-06-17 14:24:31 -07:00
parent e923434d08
commit f1b3c6b735
3 changed files with 109 additions and 100 deletions
--- a/comfyui_manager/glob/constants.py
+++ b/comfyui_manager/glob/constants.py
@@ -33,3 +33,22 @@ model_dir_name_map = {
    "unet": "diffusion_models",
    "diffusion_model": "diffusion_models",
 }
+
+# List of all model directory names used for checking installed models
+MODEL_DIR_NAMES = [
+    "checkpoints",
+    "loras",
+    "vae",
+    "text_encoders",
+    "diffusion_models",
+    "clip_vision",
+    "embeddings",
+    "diffusers",
+    "vae_approx",
+    "controlnet",
+    "gligen",
+    "upscale_models",
+    "hypernetworks",
+    "photomaker",
+    "classifiers",
+]