1580158
/
Stable-Diffusion-Webui-Civitai-Helper
mirror of https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
javascript/civitai_helper.js
View File

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

1
scripts/ch_lib/model.py
View File

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