From e140bbf49e3eb74fb20450e45576436af6c46c64 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 31 Jan 2024 13:51:36 -0800 Subject: [PATCH] Update sd_unet.py --- modules/sd_unet.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/modules/sd_unet.py b/modules/sd_unet.py index a771849c..151d682b 100644 --- a/modules/sd_unet.py +++ b/modules/sd_unet.py @@ -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}")