lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-07 16:39:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

i

Browse Source
This commit is contained in:
lllyasviel
2024-01-27 21:46:33 -08:00
parent 0db0e3ed88
commit e8175e9a90
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers_common.py
View File

@@ -190,7 +190,7 @@ def apply_refiner(cfg_denoiser, x):
unet_patcher = refiner.forge_objects.unet
model_management.load_models_gpu(
[unet_patcher],
unet_patcher.memory_required([x.shape[0]] + list(x.shape[1:])) + inference_memory)
unet_patcher.memory_required([x.shape[0]] * 2 + list(x.shape[1:])) + inference_memory)
return True

3
modules_forge/forge_sampler.py
View File

@@ -70,3 +70,6 @@ def forge_sample(self, denoiser_params, cond_scale, cond_composition):
denoised = sampling_function(model, x, timestep, uncond, cond, cond_scale, model_options, seed)
return denoised
# def prepare_sampling(unet, )