lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 18:51:31 +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): def send_model_to_cpu(m):
if m.lowvram: pass
lowvram.send_everything_to_cpu()
else:
m.to(devices.cpu)
devices.torch_gc()
def model_target_device(m): def model_target_device(m):
if lowvram.is_needed(m): return devices.device
return devices.cpu
else:
return devices.device
def send_model_to_device(m): def send_model_to_device(m):
lowvram.apply(m) pass
if not m.lowvram:
m.to(shared.device)
def send_model_to_trash(m): def send_model_to_trash(m):
m.to(device="meta") pass
devices.torch_gc()
def load_model(checkpoint_info=None, already_loaded_state_dict=None): def load_model(checkpoint_info=None, already_loaded_state_dict=None):