Update openshape/__init__.py
Browse files- 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"
|
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()
|