hengli commited on
Commit
824e0dc
·
1 Parent(s): 14eb51e

update hf link

Browse files
requirements.txt CHANGED
@@ -1,3 +1,4 @@
 
1
  torch>=2.3.1
2
  torchvision>=0.18.1
3
  numpy>=1.26.1
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
  torch>=2.3.1
3
  torchvision>=0.18.1
4
  numpy>=1.26.1
sailrecon/models/sail_recon.py CHANGED
@@ -20,7 +20,7 @@ from sailrecon.models.aggregator import Aggregator
20
  from sailrecon.utils.geometry import unproject_depth_map_to_point_map
21
  from sailrecon.utils.pose_enc import pose_encoding_to_extri_intri
22
 
23
- @spaces.GPU
24
  class SailRecon(nn.Module, PyTorchModelHubMixin):
25
  def __init__(
26
  self,
 
20
  from sailrecon.utils.geometry import unproject_depth_map_to_point_map
21
  from sailrecon.utils.pose_enc import pose_encoding_to_extri_intri
22
 
23
+ # @spaces.GPU
24
  class SailRecon(nn.Module, PyTorchModelHubMixin):
25
  def __init__(
26
  self,