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, )