Skip to content

Commit d69136d

Browse files
committed
compatible for torch >= 2.6.0
1 parent 50e0e3c commit d69136d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nerfstudio/utils/eval_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def eval_load_checkpoint(config: TrainerConfig, pipeline: Pipeline) -> Tuple[Pat
5959
load_step = config.load_step
6060
load_path = config.load_dir / f"step-{load_step:09d}.ckpt"
6161
assert load_path.exists(), f"Checkpoint {load_path} does not exist"
62-
loaded_state = torch.load(load_path, map_location="cpu")
62+
loaded_state = torch.load(load_path, map_location="cpu", weights_only=False)
6363
pipeline.load_pipeline(loaded_state["pipeline"], loaded_state["step"])
6464
CONSOLE.print(f":white_check_mark: Done loading checkpoint from {load_path}")
6565
return load_path, load_step

0 commit comments

Comments
 (0)