feat: mapping clip and checkpoint to model path (#942)

This commit is contained in:
LiGongrong 2024-08-09 20:15:40 +08:00 committed by GitHub
parent 3892e3f5e7
commit b491f51a04
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -249,8 +249,12 @@ def get_model_dir(data):
model_type = data['type'] model_type = data['type']
if model_type == "checkpoints": if model_type == "checkpoints":
base_model = folder_paths.folder_names_and_paths["checkpoints"][0][0] 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": elif model_type == "unclip":
base_model = folder_paths.folder_names_and_paths["checkpoints"][0][0] 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": elif model_type == "VAE":
base_model = folder_paths.folder_names_and_paths["vae"][0][0] base_model = folder_paths.folder_names_and_paths["vae"][0][0]
elif model_type == "lora": elif model_type == "lora":