From f3e6e6d506f75f8d3929b77fc79e6bbab4e9207b Mon Sep 17 00:00:00 2001 From: VALADI K JAGANATHAN Date: Wed, 29 Nov 2023 23:48:50 +0530 Subject: [PATCH] Update bnk_tiled_samplers.py --- py/bnk_tiled_samplers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/bnk_tiled_samplers.py b/py/bnk_tiled_samplers.py index dcb90e5..16f0d37 100644 --- a/py/bnk_tiled_samplers.py +++ b/py/bnk_tiled_samplers.py @@ -123,7 +123,7 @@ def sample_common(model, add_noise, noise_seed, tile_width, tile_height, tiling_ positive_copy = comfy.sample.convert_cond(positive) negative_copy = comfy.sample.convert_cond(negative) modelPatches, inference_memory = comfy.sample.get_additional_models(positive_copy, negative_copy, model.model_dtype()) - comfy.model_management.load_models_gpu([model] + modelPatches, memory_required=(noise.shape[0] * noise.shape[2] * noise.shape[3]) + inference_memory) + comfy.model_management.load_models_gpu([model] + modelPatches, model.memory_required(noise.shape) + inference_memory) real_model = model.model sampler = comfy.samplers.KSampler(real_model, steps=steps, device=device, sampler=sampler_name, scheduler=scheduler, denoise=denoise, model_options=model.model_options)