From 477199357f4f5f02d62857a0cf432a3ed19e6418 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sat, 13 May 2023 20:15:37 +0300
Subject: add an option to always refer to lora by filenames never refer to
 lora by an alias if multiple loras have same alias or the alias is called
 none

---
 extensions-builtin/Lora/lora.py | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'extensions-builtin/Lora/lora.py')

diff --git a/extensions-builtin/Lora/lora.py b/extensions-builtin/Lora/lora.py
index ba1293df..6fa80006 100644
--- a/extensions-builtin/Lora/lora.py
+++ b/extensions-builtin/Lora/lora.py
@@ -393,6 +393,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)
 
@@ -406,6 +408,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
 
@@ -445,6 +450,7 @@ def infotext_pasted(infotext, params):
 
 available_loras = {}
 available_lora_aliases = {}
+forbidden_lora_aliases = {}
 loaded_loras = []
 
 list_available_loras()
-- 
cgit v1.2.1