Spaces:
Build error
Build error
ChrisPreston
commited on
Commit
·
0684a78
1
Parent(s):
bccf969
Update infer_tools/infer_tool.py
Browse files
infer_tools/infer_tool.py
CHANGED
@@ -68,7 +68,7 @@ class Svc:
|
|
68 |
}
|
69 |
|
70 |
self.model_path = model_path
|
71 |
-
self.dev = torch.device("
|
72 |
|
73 |
self._ = set_hparams(config=config_name, exp_name=self.project_name, infer=True,
|
74 |
reset=True, hparams_str='', print_hparams=False)
|
|
|
68 |
}
|
69 |
|
70 |
self.model_path = model_path
|
71 |
+
self.dev = torch.device("cpu")
|
72 |
|
73 |
self._ = set_hparams(config=config_name, exp_name=self.project_name, infer=True,
|
74 |
reset=True, hparams_str='', print_hparams=False)
|