feat: mapping clip and checkpoint to model path (#942)
This commit is contained in:
@@ -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":
|
||||||
|
|||||||
Reference in New Issue
Block a user