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

Update sd_unet.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-31 13:51:36 -08:00
parent 972f87ae54
commit e140bbf49e
1 changed files with 0 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modules/sd_unet.py
View File

@@ -45,15 +45,8 @@ def apply_unet(option=None):
current_unet_option = new_option
if current_unet_option is None:
current_unet = None
if not shared.sd_model.lowvram:
shared.sd_model.model.diffusion_model.to(devices.device)
return
shared.sd_model.model.diffusion_model.to(devices.cpu)
devices.torch_gc()
current_unet = current_unet_option.create_unet()
current_unet.option = current_unet_option
print(f"Activating unet: {current_unet.option.label}")