OpenShape
/

openshape-demo-support

Model card Files Files and versions

eliphatfs commited on Dec 5, 2023

Commit

3dd5bf6

·

1 Parent(s): 24cb86c

Update openshape/init.py

Files changed (1) hide show

openshape/__init__.py +1 -1

openshape/__init__.py CHANGED Viewed

@@ -40,7 +40,7 @@ model_list = {
 def load_pc_encoder(name):
-    s = torch.load(hf_hub_download("OpenShape/" + name, "model.pt", token=True), map_location='cpu')
     model = model_list[name](s).eval()
     if torch.cuda.is_available():
         model.cuda()

 def load_pc_encoder(name):
+    s = torch.load(hf_hub_download("OpenShape/" + name, "model.pt"), map_location='cpu')
     model = model_list[name](s).eval()
     if torch.cuda.is_available():
         model.cuda()