ChrisPreston commited on
Commit
0684a78
1 Parent(s): bccf969

Update infer_tools/infer_tool.py

Browse files
Files changed (1) hide show
  1. infer_tools/infer_tool.py +1 -1
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("cuda" if torch.cuda.is_available() else "cpu")
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)