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

Run insightface on CPU and adapt the code to make it work with ZeroGPU

Browse files
Files changed (1) hide show
  1. src/utils/landmark_runner.py +1 -1
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()