From b491f51a04e2b776db5cbf0d59c421cb0ffe85ea Mon Sep 17 00:00:00 2001 From: LiGongrong Date: Fri, 9 Aug 2024 20:15:40 +0800 Subject: [PATCH] feat: mapping clip and checkpoint to model path (#942) --- glob/manager_server.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/glob/manager_server.py b/glob/manager_server.py index 92f05cb2..95ff7605 100644 --- a/glob/manager_server.py +++ b/glob/manager_server.py @@ -249,8 +249,12 @@ def get_model_dir(data): model_type = data['type'] if model_type == "checkpoints": base_model = folder_paths.folder_names_and_paths["checkpoints"][0][0] + elif model_type == "checkpoint": + base_model = folder_paths.folder_names_and_paths["checkpoints"][0][0] elif model_type == "unclip": base_model = folder_paths.folder_names_and_paths["checkpoints"][0][0] + elif model_type == "clip": + base_model = folder_paths.folder_names_and_paths["clip"][0][0] elif model_type == "VAE": base_model = folder_paths.folder_names_and_paths["vae"][0][0] elif model_type == "lora":