soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

Fix LoRA infotext (#406)

Browse Source
This commit is contained in:
catboxanon 2024-02-25 22:41:42 -05:00 committed by GitHub
parent beb6e76135
commit b458e42096
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
extensions-builtin/Lora/networks.py
View File

@ -27,7 +27,10 @@ def assign_network_names_to_compvis_modules(sd_model):
def load_network(name, network_on_disk):
pass
net = network.Network(name, network_on_disk)
net.mtime = os.path.getmtime(network_on_disk.filename)
return net
def purge_networks_from_memory():
@ -41,11 +44,23 @@ def load_networks(names, te_multipliers=None, unet_multipliers=None, dyn_dims=No
if current_sd is None:
return
loaded_networks.clear()
networks_on_disk = [available_networks.get(name, None) if name.lower() in forbidden_network_aliases else available_network_aliases.get(name, None) for name in names]
if any(x is None for x in networks_on_disk):
list_available_networks()
networks_on_disk = [available_networks.get(name, None) if name.lower() in forbidden_network_aliases else available_network_aliases.get(name, None) for name in names]
for i, (network_on_disk, name) in enumerate(zip(networks_on_disk, names)):
try:
net = load_network(name, network_on_disk)
except Exception as e:
errors.display(e, f"loading network {network_on_disk.filename}")
continue
net.mentioned_name = name
network_on_disk.read_hash()
loaded_networks.append(net)
compiled_lora_targets = []
for a, b, c in zip(networks_on_disk, unet_multipliers, te_multipliers):
compiled_lora_targets.append([a.filename, b, c])