From 4cf45c71c2553a76ce3270ffbed18ac46ac3f364 Mon Sep 17 00:00:00 2001
From: Robert Wojciechowski <hello@robertvoy.com>
Date: Thu, 11 Jun 2026 12:39:06 +0000
Subject: [PATCH] fix: keep local LTX tiling in execution context

---
 nodes/ltx_tiled_sampler.py      | 73 ++++++++++++++++++++++++++++++---
 tests/test_ltx_tiled_sampler.py | 28 +++++++++++++
 2 files changed, 95 insertions(+), 6 deletions(-)

diff --git a/nodes/ltx_tiled_sampler.py b/nodes/ltx_tiled_sampler.py
index 16d1e74..2d275f7 100644
--- a/nodes/ltx_tiled_sampler.py
+++ b/nodes/ltx_tiled_sampler.py
@@ -476,6 +476,54 @@ async def _ltx_master_process(
     return samples, denoised
 
 
+def _ltx_local_tiled_process(
+    noise,
+    guider,
+    sampler,
+    sigmas,
+    latent_tensor,
+    denoise_mask,
+    n_h_tiles,
+    n_w_tiles,
+    tile_overlap,
+):
+    """Process all LTX latent tiles locally in the current ComfyUI execution context.
+
+    This path is used when no workers are enabled. Keeping it synchronous avoids
+    crossing ComfyUI's execution/inference context boundary while still using the
+    same tile extraction and blend logic as the distributed master/worker path.
+    """
+    tiles = _build_tile_queue(latent_tensor.shape, n_h_tiles, n_w_tiles, tile_overlap)
+    full_noise = noise.generate_noise({"samples": latent_tensor})
+    output_sum = torch.zeros_like(latent_tensor)
+    weight_sum = torch.zeros_like(latent_tensor)
+    denoised_sum = None
+
+    for tile in tiles:
+        tile_latent = _tile_latent(latent_tensor, tile)
+        tile_noise = _tile_latent(full_noise, tile)
+        tile_mask = _tile_mask(denoise_mask, tile)
+        tile_samples, tile_denoised = _sample_tile(guider, tile_noise, tile_latent, sampler, sigmas, tile_mask)
+        if denoised_sum is None and tile_denoised is not None:
+            denoised_sum = torch.zeros_like(latent_tensor)
+        _blend_tile_into_sums(
+            tile,
+            tile_samples,
+            tile_denoised,
+            output_sum,
+            weight_sum,
+            denoised_sum,
+            n_h_tiles,
+            n_w_tiles,
+            tile_overlap,
+        )
+
+    weight_sum = torch.where(weight_sum == 0, torch.ones_like(weight_sum), weight_sum)
+    samples = output_sum / weight_sum
+    denoised = denoised_sum / weight_sum if denoised_sum is not None else None
+    return samples, denoised
+
+
 class LTXTiledSamplerDistributed:
     """Distributed wrapper around the LTX tiled sampler."""
 
@@ -583,9 +631,8 @@ class LTXTiledSamplerDistributed:
             placeholder = rebuild_latent(latent_tensor)
             return (placeholder, placeholder, placeholder)
 
-        samples, denoised = run_async_in_server_loop(
-            _ltx_master_process(
-                self,
+        if not self.enabled_worker_ids:
+            samples, denoised = _ltx_local_tiled_process(
                 noise,
                 guider,
                 sampler,
@@ -595,9 +642,23 @@ class LTXTiledSamplerDistributed:
                 int(h_tiles),
                 int(w_tiles),
                 int(overlap),
-            ),
-            timeout=None,
-        )
+            )
+        else:
+            samples, denoised = run_async_in_server_loop(
+                _ltx_master_process(
+                    self,
+                    noise,
+                    guider,
+                    sampler,
+                    sigmas,
+                    latent_tensor,
+                    mask_tensor,
+                    int(h_tiles),
+                    int(w_tiles),
+                    int(overlap),
+                ),
+                timeout=None,
+            )
         samples_latent = rebuild_latent(samples)
         denoised_latent = rebuild_latent(denoised if denoised is not None else samples)
         return (samples_latent, denoised_latent, samples_latent)
diff --git a/tests/test_ltx_tiled_sampler.py b/tests/test_ltx_tiled_sampler.py
index f29763c..00aa71c 100644
--- a/tests/test_ltx_tiled_sampler.py
+++ b/tests/test_ltx_tiled_sampler.py
@@ -125,5 +125,33 @@ class LtxTiledSamplerTests(unittest.TestCase):
         self.assertIn("LTXTiledSamplerDistributed", ltx.NODE_CLASS_MAPPINGS)
 
 
+    def test_local_tiled_process_runs_without_async_master_path(self):
+        class _Noise:
+            def generate_noise(self, latent):
+                return torch.ones_like(latent['samples']) * 0.25
+
+        class _Guider:
+            def sample(self, noise, latent_image, sampler, sigmas, denoise_mask=None, disable_pbar=True):
+                return latent_image + noise
+
+        latent = torch.zeros((1, 1, 2, 4, 4), dtype=torch.float32)
+
+        samples, denoised = ltx._ltx_local_tiled_process(
+            _Noise(),
+            _Guider(),
+            sampler=object(),
+            sigmas=torch.tensor([1.0, 0.0]),
+            latent_tensor=latent,
+            denoise_mask=None,
+            n_h_tiles=2,
+            n_w_tiles=2,
+            tile_overlap=0,
+        )
+
+        self.assertIsNone(denoised)
+        self.assertEqual(tuple(samples.shape), tuple(latent.shape))
+        self.assertTrue(torch.allclose(samples, torch.full_like(latent, 0.25)))
+
+
 if __name__ == "__main__":
     unittest.main()