From 7cd1d97477f09b39b0a456d65d331e8fad673434 Mon Sep 17 00:00:00 2001 From: VALADI K JAGANATHAN Date: Mon, 27 May 2024 10:49:54 +0530 Subject: [PATCH] Update bnk_tiled_samplers.py changes in line 131 check same --- py/bnk_tiled_samplers.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/py/bnk_tiled_samplers.py b/py/bnk_tiled_samplers.py index 3c5d011..084bba3 100644 --- a/py/bnk_tiled_samplers.py +++ b/py/bnk_tiled_samplers.py @@ -128,8 +128,10 @@ def sample_common(model, add_noise, noise_seed, tile_width, tile_height, tiling_ modelPatches, inference_memory = comfy.sample.get_additional_models(positive_copy, negative_copy, model.model_dtype()) 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) +#smaller changes + model_patcher = comfy.model_patcher.ModelPatcher(model.model, load_device=device, offload_device=comfy.model_management.unet_offload_device()) + sampler = comfy.samplers.KSampler(model_patcher, steps=steps, device=device, sampler=sampler_name, scheduler=scheduler, denoise=denoise, model_options=model.model_options) + #sampler = comfy.samplers.KSampler(real_model, steps=steps, device=device, sampler=sampler_name, scheduler=scheduler, denoise=denoise, model_options=model.model_options) if tiling_strategy != 'padded': if noise_mask is not None: