diff --git a/javascript/civitai_helper.js b/javascript/civitai_helper.js index 4454d0d..686731e 100644 --- a/javascript/civitai_helper.js +++ b/javascript/civitai_helper.js @@ -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; } diff --git a/scripts/ch_lib/model.py b/scripts/ch_lib/model.py index 729012d..e1f3ae9 100644 --- a/scripts/ch_lib/model.py +++ b/scripts/ch_lib/model.py @@ -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")