aboutsummaryrefslogtreecommitdiffstats
path: root/extensions-builtin/Lora/lora.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-05-13 17:26:13 +0000
committerAUTOMATIC <16777216c@gmail.com>2023-05-13 17:26:13 +0000
commite8eea1bb7a8454c5b2453695fff23d4afb88f2b1 (patch)
treed402d2443ef59ec6ad8743efd3b8533a693551b7 /extensions-builtin/Lora/lora.py
parentd70c3a807b6904efe35c1eeb29997121a28143f2 (diff)
parent2053745c8fc29a0d3c1bbfa07858fb0b21b32e6d (diff)
downloadstable-diffusion-webui-gfx803-e8eea1bb7a8454c5b2453695fff23d4afb88f2b1.tar.gz
stable-diffusion-webui-gfx803-e8eea1bb7a8454c5b2453695fff23d4afb88f2b1.tar.bz2
stable-diffusion-webui-gfx803-e8eea1bb7a8454c5b2453695fff23d4afb88f2b1.zip
Merge branch 'release_candidate' into dev
Diffstat (limited to 'extensions-builtin/Lora/lora.py')
-rw-r--r--extensions-builtin/Lora/lora.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py
index 7b56136f..1a5b3c2d 100644
--- a/extensions-builtin/Lora/lora.py
+++ b/extensions-builtin/Lora/lora.py
@@ -392,6 +392,8 @@ def lora_MultiheadAttention_load_state_dict(self, *args, **kwargs):
def list_available_loras():
available_loras.clear()
available_lora_aliases.clear()
+ forbidden_lora_aliases.clear()
+ forbidden_lora_aliases.update({"none": 1})
os.makedirs(shared.cmd_opts.lora_dir, exist_ok=True)
@@ -405,6 +407,9 @@ def list_available_loras():
available_loras[name] = entry
+ if entry.alias in available_lora_aliases:
+ forbidden_lora_aliases[entry.alias.lower()] = 1
+
available_lora_aliases[name] = entry
available_lora_aliases[entry.alias] = entry
@@ -444,6 +449,7 @@ def infotext_pasted(infotext, params):
available_loras = {}
available_lora_aliases = {}
+forbidden_lora_aliases = {}
loaded_loras = []
list_available_loras()