Fix torch.compile for torch 2.1 (nerfstudio-project#2577)

Yosshi999 · Nov 3, 2023 · a58ec45 · a58ec45
1 parent 850d515
commit a58ec45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,