From da13b6b827d9291718a51fc9cc4ea54266a43f23 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Fri, 20 Dec 2024 18:02:12 -0500 Subject: [PATCH] Get rid of meshgrid warning. --- 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 51ce5058..c58dfb20 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='ij') grid = torch.stack(grid, dim=0) grid = grid.unsqueeze(0).repeat(batch_size, 1, 1, 1, 1)