From a077975c3e2c6d221a4a26340f0103dacc1df0f5 Mon Sep 17 00:00:00 2001 From: VALADI K JAGANATHAN Date: Sun, 26 Nov 2023 12:27:01 +0530 Subject: [PATCH] Update bnk_tiled_samplers.py memory management changed in comfy original repo so the code got changed. --- 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 09636ce..dcb90e5 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, comfy.model_management.batch_area_memory(noise.shape[0] * noise.shape[2] * noise.shape[3]) + inference_memory) + comfy.model_management.load_models_gpu([model] + modelPatches, memory_required=(noise.shape[0] * noise.shape[2] * noise.shape[3]) + 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)