OpenShape
/

openshape-demo-support

eliphatfs commited on Dec 5, 2023

Commit

70dbc29

1 Parent(s): 5e2636e

Update openshape/demo/retrieval.py

Files changed (1) hide show

openshape/demo/retrieval.py CHANGED Viewed

@@ -24,7 +24,7 @@ meta = json.load(
 # }
 meta = {x['u']: x for x in meta['entries']}
 deser = torch.load(
-    hf_hub_download("OpenShape/openshape-objaverse-embeddings", "objaverse.pt", token=True, repo_type='dataset'), map_location='cpu'
 )
 us = deser['us']
 feats = deser['feats']

 # }
 meta = {x['u']: x for x in meta['entries']}
 deser = torch.load(
+    hf_hub_download("OpenShape/openshape-objaverse-embeddings", "objaverse.pt", repo_type='dataset'), map_location='cpu'
 )
 us = deser['us']
 feats = deser['feats']