From ca9b1b18e692c5c1037ec755fa6c6136343cedf4 Mon Sep 17 00:00:00 2001 From: Miklos Nagy Date: Fri, 20 Dec 2024 19:29:20 +0100 Subject: [PATCH] Update symmetric_patchifier.py fix torch.compile error with meshgrid being called without indexing kwarg --- comfy/ldm/lightricks/symmetric_patchifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/ldm/lightricks/symmetric_patchifier.py b/comfy/ldm/lightricks/symmetric_patchifier.py index 51ce50589..4df9016a3 100644 --- a/comfy/ldm/lightricks/symmetric_patchifier.py +++ b/comfy/ldm/lightricks/symmetric_patchifier.py @@ -53,7 +53,7 @@ class Patchifier(ABC): grid_h = torch.arange(h, dtype=torch.float32, device=device) grid_w = torch.arange(w, dtype=torch.float32, device=device) grid_f = torch.arange(f, dtype=torch.float32, device=device) - grid = torch.meshgrid(grid_f, grid_h, grid_w) + grid = torch.meshgrid(grid_f, grid_h, grid_w, indexing=None) grid = torch.stack(grid, dim=0) grid = grid.unsqueeze(0).repeat(batch_size, 1, 1, 1, 1)