Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
20afb24
1
Parent(s):
0e9358c
Run insightface on CPU and adapt the code to make it work with ZeroGPU
Browse files
src/utils/landmark_runner.py
CHANGED
@@ -27,7 +27,7 @@ class LandmarkRunner(object):
|
|
27 |
"""landmark runner"""
|
28 |
def __init__(self, **kwargs):
|
29 |
ckpt_path = kwargs.get('ckpt_path')
|
30 |
-
onnx_provider = kwargs.get('onnx_provider', 'cuda') # 默认用cuda
|
31 |
device_id = kwargs.get('device_id', 0)
|
32 |
self.dsize = kwargs.get('dsize', 224)
|
33 |
self.timer = Timer()
|
|
|
27 |
"""landmark runner"""
|
28 |
def __init__(self, **kwargs):
|
29 |
ckpt_path = kwargs.get('ckpt_path')
|
30 |
+
onnx_provider = 'cpu' #run on cpu for it to work with ZeroGPU // kwargs.get('onnx_provider', 'cuda') # 默认用cuda
|
31 |
device_id = kwargs.get('device_id', 0)
|
32 |
self.dsize = kwargs.get('dsize', 224)
|
33 |
self.timer = Timer()
|