soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

Update forge_sampler.py

Browse Source
This commit is contained in:
lllyasviel 2024-01-27 22:13:29 -08:00
parent d51e81483b
commit ba03d74920
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules_forge/forge_sampler.py
View File

@ -77,8 +77,12 @@ def sampling_prepare(unet, x):
B, C, H, W = x.shape
unet_inference_memory = unet.memory_required([B * 2, C, H, W])
additional_inference_memory = unet.controlnet_linked_list.inference_memory_requirements(unet.model_dtype())
additional_model_patchers = unet.get_models()
additional_inference_memory = 0
additional_model_patchers = []
if unet.controlnet_linked_list is not None:
additional_inference_memory += unet.controlnet_linked_list.inference_memory_requirements(unet.model_dtype())
additional_model_patchers += unet.controlnet_linked_list.get_models()
model_management.load_models_gpu(
models=[unet] + additional_model_patchers,