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

View File

@@ -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":