From e8175e9a90cd78bf1dcc44a7f3ab3f9ca2f44c26 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sat, 27 Jan 2024 21:46:33 -0800 Subject: [PATCH] i --- modules/sd_samplers_common.py | 2 +- modules_forge/forge_sampler.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/modules/sd_samplers_common.py b/modules/sd_samplers_common.py index 5b1311f9..ec551257 100644 --- a/modules/sd_samplers_common.py +++ b/modules/sd_samplers_common.py @@ -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 diff --git a/modules_forge/forge_sampler.py b/modules_forge/forge_sampler.py index 7028ef62..f2e84a83 100644 --- a/modules_forge/forge_sampler.py +++ b/modules_forge/forge_sampler.py @@ -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, )