fix lora name matching

This commit is contained in:
Vladimir Mandic
2024-09-16 16:33:16 -04:00
parent 6df232f59a
commit e0904fbc43
2 changed files with 13 additions and 3 deletions
+2
View File
@@ -512,6 +512,8 @@ def list_available_networks():
try:
entry = network.NetworkOnDisk(name, filename)
available_networks[entry.name] = entry
if '.' in entry.name:
available_networks[entry.name.replace('.', '_')] = entry
if entry.alias in available_network_aliases:
forbidden_network_aliases[entry.alias.lower()] = 1
if shared.opts.lora_preferred_name == 'filename':
@@ -6,6 +6,9 @@ import networks
from modules import shared, ui_extra_networks
debug = os.environ.get('SD_LOAD_DEBUG', None) is not None
class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
def __init__(self):
super().__init__('Lora')
@@ -16,6 +19,10 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
def create_item(self, name):
l = networks.available_networks.get(name)
if l is None:
shared.log.warning(f'Networks: type=lora registered={len(list(networks.available_networks))} file="{name}" not registered')
print(networks.available_networks)
return None
try:
# path, _ext = os.path.splitext(l.filename)
name = os.path.splitext(os.path.relpath(l.filename, shared.cmd_opts.lora_dir))[0]
@@ -103,9 +110,10 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
return item
except Exception as e:
shared.log.debug(f"Networks error: type=lora file={name} {e}")
from modules import errors
errors.display('e', 'Lora')
shared.log.error(f"Networks: type=lora file={name} {e}")
if debug:
from modules import errors
errors.display('e', 'Lora')
return None
def list_items(self):