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

feat: Add support for LyCoris tab

pull/186/head
Christoph Hess 2023-06-06 01:48:37 +02:00
parent 920ca3267f
commit 199cfba737
2 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
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
@ -432,7 +432,7 @@ onUiLoaded(() => {
//get active extratab //get active extratab
const active_extra_tab = Array.from(get_uiCurrentTabContent().querySelectorAll('.extra-network-cards,.extra-network-thumbs')) const active_extra_tab = Array.from(get_uiCurrentTabContent().querySelectorAll('.extra-network-cards,.extra-network-thumbs'))
.find(el => el.closest('.tabitem').style.display === 'block') .find(el => el.closest('.tabitem').style.display === 'block')
?.id.match(/^(txt2img|img2img)_(.+)_cards$/)[2] ?.id.match(/^(txt2img|img2img|tab_twoshot_mask)_(.+)_cards$/)[2]
console.log("found active tab: " + active_extra_tab); console.log("found active tab: " + active_extra_tab);
@ -450,6 +450,9 @@ onUiLoaded(() => {
case "lora": case "lora":
active_extra_tab_type = "lora"; active_extra_tab_type = "lora";
break; break;
case "lycoris":
active_extra_tab_type = "lycoris";
break;
} }
@ -487,10 +490,16 @@ onUiLoaded(() => {
extra_network_id = tab_prefix+"_"+js_model_type+"_"+cardid_suffix; extra_network_id = tab_prefix+"_"+js_model_type+"_"+cardid_suffix;
// console.log("searching extra_network_node: " + extra_network_id); // console.log("searching extra_network_node: " + extra_network_id);
extra_network_node = gradioApp().getElementById(extra_network_id); extra_network_node = gradioApp().getElementById(extra_network_id);
// check if extr network is under thumbnail mode let lycoris_node = gradioApp().getElementById(tab_prefix+"_lycoris_"+cardid_suffix);
let is_lycoris = !!lycoris_node.offsetWidth;
if (is_lycoris) {
extra_network_node = lycoris_node;
model_type = "lycoris";
}
// check if extra network is under thumbnail mode
is_thumb_mode = false is_thumb_mode = false
if (extra_network_node) { if (extra_network_node) {
if (extra_network_node.className == "extra-network-thumbs") { if (extra_network_node.className == "extra-network-thumbs" || extra_network_node.className == "extra-network-cards") {
console.log(extra_network_id + " is in thumbnail mode"); console.log(extra_network_id + " is in thumbnail mode");
is_thumb_mode = true; is_thumb_mode = true;
// if (!ch_show_btn_on_thumb) {continue;} // if (!ch_show_btn_on_thumb) {continue;}

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")