OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-04-13 16:29:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #15600 from AUTOMATIC1111/fix-corrupt-model-loop

Browse Source 
Fix corrupt model initial load loop
This commit is contained in:
AUTOMATIC1111 2024-06-08 10:12:45 +03:00 committed by GitHub
commit 15245d9d5e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 17 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
extensions-builtin/Lora/ui_extra_networks_lora.py
View File

@ -60,6 +60,7 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
else: else:
sd_version = lora_on_disk.sd_version sd_version = lora_on_disk.sd_version
if shared.sd_model is not None: # still show LoRA in case an error occurs during initial model loading
if shared.opts.lora_show_all or not enable_filter: if shared.opts.lora_show_all or not enable_filter:
pass pass
elif sd_version == network.SdVersion.Unknown: elif sd_version == network.SdVersion.Unknown:

1
modules/sd_models.py
View File

@ -659,6 +659,7 @@ def get_empty_cond(sd_model):
def send_model_to_cpu(m): def send_model_to_cpu(m):
if m is not None:
if m.lowvram: if m.lowvram:
lowvram.send_everything_to_cpu() lowvram.send_everything_to_cpu()
else: else: