Skip to content

Commit 36187d5

Browse files
liruilong940607Ruilong Li
and
Ruilong Li
authored
torch.compile -> torch_compile from misc (#3202)
Co-authored-by: Ruilong Li <[email protected]>
1 parent d92d76b commit 36187d5

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

nerfstudio/models/splatfacto.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
from nerfstudio.model_components import renderers
4747
from nerfstudio.models.base_model import Model, ModelConfig
4848
from nerfstudio.utils.colors import get_color
49+
from nerfstudio.utils.misc import torch_compile
4950
from nerfstudio.utils.rich_utils import CONSOLE
5051

5152

@@ -99,7 +100,7 @@ def resize_image(image: torch.Tensor, d: int):
99100
return tf.conv2d(image.permute(2, 0, 1)[:, None, ...], weight, stride=d).squeeze(1).permute(1, 2, 0)
100101

101102

102-
@torch.compile()
103+
@torch_compile()
103104
def get_viewmat(optimized_camera_to_world):
104105
"""
105106
function that converts c2w to gsplat world2camera matrix, using compile for some speed

0 commit comments

Comments
 (0)