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

Merge branch 'Add-LyCORIS-parsing' of https://github.com/goldmojo/Stable-Diffusion-Webui-Civitai-Helper into Add-LyCORIS-parsing

pull/107/head
Goldmojo 2023-06-25 19:36:42 +02:00
parent 53ca8030f1
commit 2aa71a6df9
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/ch_lib/civitai.py
View File

@ -23,7 +23,7 @@ model_type_dict = {
"TextualInversion": "ti", "TextualInversion": "ti",
"Hypernetwork": "hyper", "Hypernetwork": "hyper",
"LORA": "lora", "LORA": "lora",
"LoCon": "lora", "LoCon": "lycoris",
} }

6
scripts/ch_lib/model.py
View File

@ -17,7 +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"), "lycoris": os.path.join(root_path, "models", "LyCORIS"),
} }
exts = (".bin", ".pt", ".safetensors", ".ckpt") exts = (".bin", ".pt", ".safetensors", ".ckpt")
@ -43,8 +43,8 @@ def get_custom_model_folder():
if shared.cmd_opts.lora_dir and os.path.isdir(shared.cmd_opts.lora_dir): if shared.cmd_opts.lora_dir and os.path.isdir(shared.cmd_opts.lora_dir):
folders["lora"] = shared.cmd_opts.lora_dir folders["lora"] = shared.cmd_opts.lora_dir
if shared.cmd_opts.lyco_dir and os.path.isdir(shared.cmd_opts.lyco_dir):
folders["lycoris"] = shared.cmd_opts.lyco_dir
# write model info to file # write model info to file