From 85677d310fcb8700b284225de646c63bb02b83de Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jon=C3=A1=C5=A1=20Kulh=C3=A1nek?= Date: Thu, 2 Nov 2023 09:28:11 +0100 Subject: [PATCH] Fix torch.compile for torch 2.1 (#2577) --- nerfstudio/cameras/camera_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/cameras/camera_utils.py b/nerfstudio/cameras/camera_utils.py index 8c527b9db..40419528a 100644 --- a/nerfstudio/cameras/camera_utils.py +++ b/nerfstudio/cameras/camera_utils.py @@ -407,7 +407,7 @@ def _compute_residual_and_jacobian( return fx, fy, fx_x, fx_y, fy_x, fy_y -@torch_compile(dynamic=True, mode="reduce-overhead", backend="eager") +@torch_compile(dynamic=True, mode="reduce-overhead") def radial_and_tangential_undistort( coords: torch.Tensor, distortion_params: torch.Tensor,