lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-09 01:19:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sd_models.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 08:01:48 -08:00
parent 54db5ee40c
commit 618a479db7
1 changed files with 4 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
modules/sd_models.py
View File

@@ -542,31 +542,19 @@ def get_empty_cond(sd_model):
def send_model_to_cpu(m):
if m.lowvram:
lowvram.send_everything_to_cpu()
else:
m.to(devices.cpu)
devices.torch_gc()
pass
def model_target_device(m):
if lowvram.is_needed(m):
return devices.cpu
else:
return devices.device
return devices.device
def send_model_to_device(m):
lowvram.apply(m)
if not m.lowvram:
m.to(shared.device)
pass
def send_model_to_trash(m):
m.to(device="meta")
devices.torch_gc()
pass
def load_model(checkpoint_info=None, already_loaded_state_dict=None):