Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
dfb1c4a
1
Parent(s):
20afb24
change it in cropper.py too
Browse files- 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='
|
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=["
|
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()
|