Got 'Add Triggerwords to Prompt/Use Prompt' to work with LyCORIS folder

pull/107/head
nosdregamon 2023-06-25 01:22:42 +02:00
parent 920ca3267f
commit 53ca8030f1
2 changed files with 8 additions and 1 deletions

View File

@ -338,7 +338,7 @@ onUiLoaded(() => {
// get all extra network tabs
let tab_prefix_list = ["txt2img", "img2img"];
let model_type_list = ["textual_inversion", "hypernetworks", "checkpoints", "lora"];
let model_type_list = ["textual_inversion", "hypernetworks", "checkpoints", "lora", "lycoris"];
let cardid_suffix = "cards";
//get init py msg
@ -449,6 +449,9 @@ onUiLoaded(() => {
break;
case "lora":
active_extra_tab_type = "lora";
break;
case "lycoris":
active_extra_tab_type = "lycoris";
break;
}
@ -467,6 +470,9 @@ onUiLoaded(() => {
break;
case "lora":
model_type = "lora";
break;
case "lycoris":
model_type = "lycoris";
break;
}

View File

@ -17,6 +17,7 @@ folders = {
"hyper": os.path.join(root_path, "models", "hypernetworks"),
"ckp": os.path.join(root_path, "models", "Stable-diffusion"),
"lora": os.path.join(root_path, "models", "Lora"),
"lycoris": os.path.join(root_path, "models", "LyCORIS"),
}
exts = (".bin", ".pt", ".safetensors", ".ckpt")