levihsu commited on
Commit
6869802
1 Parent(s): a29c967

Update preprocess/openpose/annotator/openpose/body.py

Browse files
preprocess/openpose/annotator/openpose/body.py CHANGED
@@ -20,8 +20,8 @@ from .model import bodypose_model
20
  class Body(object):
21
  def __init__(self, model_path):
22
  self.model = bodypose_model()
23
- if torch.cuda.is_available():
24
- self.model = self.model.cuda()
25
  # print('cuda')
26
  model_dict = util.transfer(self.model, torch.load(model_path))
27
  self.model.load_state_dict(model_dict)
 
20
  class Body(object):
21
  def __init__(self, model_path):
22
  self.model = bodypose_model()
23
+ # if torch.cuda.is_available():
24
+ # self.model = self.model.cuda()
25
  # print('cuda')
26
  model_dict = util.transfer(self.model, torch.load(model_path))
27
  self.model.load_state_dict(model_dict)