eliphatfs commited on
Commit
3dd5bf6
1 Parent(s): 24cb86c

Update openshape/__init__.py

Browse files
Files changed (1) hide show
  1. openshape/__init__.py +1 -1
openshape/__init__.py CHANGED
@@ -40,7 +40,7 @@ model_list = {
40
 
41
 
42
  def load_pc_encoder(name):
43
- s = torch.load(hf_hub_download("OpenShape/" + name, "model.pt", token=True), map_location='cpu')
44
  model = model_list[name](s).eval()
45
  if torch.cuda.is_available():
46
  model.cuda()
 
40
 
41
 
42
  def load_pc_encoder(name):
43
+ s = torch.load(hf_hub_download("OpenShape/" + name, "model.pt"), map_location='cpu')
44
  model = model_list[name](s).eval()
45
  if torch.cuda.is_available():
46
  model.cuda()