lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-13 17:09:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

allow preserve memory

Browse Source
This commit is contained in:
lllyasviel
2024-01-31 21:57:40 -08:00
parent df5e8c520d
commit a38313368f
2 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules_forge/forge_sampler.py
View File

@@ -86,7 +86,7 @@ def forge_sample(self, denoiser_params, cond_scale, cond_composition):
def sampling_prepare(unet, x):
B, C, H, W = x.shape
unet_inference_memory = unet.memory_required([B * 2, C, H, W])
unet_inference_memory = unet.memory_required([B * 2, C, H, W]) + unet.extra_preserved_memory
additional_inference_memory = 0
additional_model_patchers = []