diff options
author | AUTOMATIC <16777216c@gmail.com> | 2023-05-13 17:26:13 +0000 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-13 17:26:13 +0000 |
commit | e8eea1bb7a8454c5b2453695fff23d4afb88f2b1 (patch) | |
tree | d402d2443ef59ec6ad8743efd3b8533a693551b7 /extensions-builtin/Lora/lora.py | |
parent | d70c3a807b6904efe35c1eeb29997121a28143f2 (diff) | |
parent | 2053745c8fc29a0d3c1bbfa07858fb0b21b32e6d (diff) | |
download | stable-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.py | 6 |
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()
|