multimodalart HF staff commited on
Commit
dfb1c4a
1 Parent(s): 20afb24

change it in cropper.py too

Browse files
Files changed (1) hide show
  1. src/utils/cropper.py +2 -2
src/utils/cropper.py CHANGED
@@ -36,7 +36,7 @@ class Cropper(object):
36
  device_id = kwargs.get('device_id', 0)
37
  self.landmark_runner = LandmarkRunner(
38
  ckpt_path=make_abs_path('../../pretrained_weights/liveportrait/landmark.onnx'),
39
- onnx_provider='cuda',
40
  device_id=device_id
41
  )
42
  self.landmark_runner.warmup()
@@ -44,7 +44,7 @@ class Cropper(object):
44
  self.face_analysis_wrapper = FaceAnalysisDIY(
45
  name='buffalo_l',
46
  root=make_abs_path('../../pretrained_weights/insightface'),
47
- providers=["CUDAExecutionProvider"]
48
  )
49
  self.face_analysis_wrapper.prepare(ctx_id=device_id, det_size=(512, 512))
50
  self.face_analysis_wrapper.warmup()
 
36
  device_id = kwargs.get('device_id', 0)
37
  self.landmark_runner = LandmarkRunner(
38
  ckpt_path=make_abs_path('../../pretrained_weights/liveportrait/landmark.onnx'),
39
+ onnx_provider='cpu',
40
  device_id=device_id
41
  )
42
  self.landmark_runner.warmup()
 
44
  self.face_analysis_wrapper = FaceAnalysisDIY(
45
  name='buffalo_l',
46
  root=make_abs_path('../../pretrained_weights/insightface'),
47
+ providers=["CPUExecutionProvider"]
48
  )
49
  self.face_analysis_wrapper.prepare(ctx_id=device_id, det_size=(512, 512))
50
  self.face_analysis_wrapper.warmup()