adaface-neurips commited on
Commit
a125540
·
1 Parent(s): a9ea043

Fix onnx option bug

Browse files
Files changed (1) hide show
  1. adaface/face_id_to_ada_prompt.py +4 -4
adaface/face_id_to_ada_prompt.py CHANGED
@@ -658,8 +658,8 @@ class Arc2Face_ID2AdaPrompt(FaceID2AdaPrompt):
658
  else:
659
  device_id = device.index
660
  self.face_app = FaceAnalysis(name='antelopev2', root='models/insightface',
661
- providers=['CUDAExecutionProvider', {'enable_fp16': True}],
662
- provider_options=[{"device_id": str(device_id)}])
663
  self.face_app.prepare(ctx_id=device_id, det_size=(512, 512))
664
 
665
  self.device = device
@@ -803,8 +803,8 @@ class ConsistentID_ID2AdaPrompt(FaceID2AdaPrompt):
803
  else:
804
  device_id = device.index
805
  self.face_app = FaceAnalysis(name='buffalo_l', root='models/insightface',
806
- providers=['CUDAExecutionProvider', {'enable_fp16': True}],
807
- provider_options=[{"device_id": str(device_id)}])
808
  self.face_app.prepare(ctx_id=device_id, det_size=(512, 512))
809
 
810
  self.device = device
 
658
  else:
659
  device_id = device.index
660
  self.face_app = FaceAnalysis(name='antelopev2', root='models/insightface',
661
+ providers=['CUDAExecutionProvider'],
662
+ provider_options=[{"device_id": device_id, 'enable_fp16': True}])
663
  self.face_app.prepare(ctx_id=device_id, det_size=(512, 512))
664
 
665
  self.device = device
 
803
  else:
804
  device_id = device.index
805
  self.face_app = FaceAnalysis(name='buffalo_l', root='models/insightface',
806
+ providers=['CUDAExecutionProvider'],
807
+ provider_options=[{"device_id": device_id, 'enable_fp16': True}])
808
  self.face_app.prepare(ctx_id=device_id, det_size=(512, 512))
809
 
810
  self.device = device