From d69136d198beea473aaa148cbd5114d01c2f847e Mon Sep 17 00:00:00 2001 From: junlinp Date: Tue, 9 Sep 2025 14:52:31 +0800 Subject: [PATCH 1/2] compatible for torch >= 2.6.0 --- nerfstudio/utils/eval_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/utils/eval_utils.py b/nerfstudio/utils/eval_utils.py index 11a8b23416..8d07fe842a 100644 --- a/nerfstudio/utils/eval_utils.py +++ b/nerfstudio/utils/eval_utils.py @@ -59,7 +59,7 @@ def eval_load_checkpoint(config: TrainerConfig, pipeline: Pipeline) -> Tuple[Pat load_step = config.load_step load_path = config.load_dir / f"step-{load_step:09d}.ckpt" assert load_path.exists(), f"Checkpoint {load_path} does not exist" - loaded_state = torch.load(load_path, map_location="cpu") + loaded_state = torch.load(load_path, map_location="cpu", weights_only=False) pipeline.load_pipeline(loaded_state["pipeline"], loaded_state["step"]) CONSOLE.print(f":white_check_mark: Done loading checkpoint from {load_path}") return load_path, load_step From 2b2a029f54ee30fd2e682f6c5a7e84c6afd71fdc Mon Sep 17 00:00:00 2001 From: junlinp Date: Wed, 10 Dec 2025 18:17:43 +0800 Subject: [PATCH 2/2] remove torch.compile for get_viewmat. --- nerfstudio/models/splatfacto.py | 1 - 1 file changed, 1 deletion(-) diff --git a/nerfstudio/models/splatfacto.py b/nerfstudio/models/splatfacto.py index 475cc5c388..6f43fba5db 100644 --- a/nerfstudio/models/splatfacto.py +++ b/nerfstudio/models/splatfacto.py @@ -62,7 +62,6 @@ def resize_image(image: torch.Tensor, d: int): return tf.conv2d(image.permute(2, 0, 1)[:, None, ...], weight, stride=d).squeeze(1).permute(1, 2, 0) -@torch_compile() def get_viewmat(optimized_camera_to_world): """ function that converts c2w to gsplat world2camera matrix, using compile for some speed