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: