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

Fix onnx option again

Browse files
Files changed (1) hide show
  1. adaface/face_id_to_ada_prompt.py +4 -2
adaface/face_id_to_ada_prompt.py CHANGED
@@ -659,7 +659,8 @@ class Arc2Face_ID2AdaPrompt(FaceID2AdaPrompt):
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
@@ -804,7 +805,8 @@ class ConsistentID_ID2AdaPrompt(FaceID2AdaPrompt):
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
 
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,
663
+ 'gpu_mem_limit': 2 * 1024 * 1024 * 1024}])
664
  self.face_app.prepare(ctx_id=device_id, det_size=(512, 512))
665
 
666
  self.device = device
 
805
  device_id = device.index
806
  self.face_app = FaceAnalysis(name='buffalo_l', root='models/insightface',
807
  providers=['CUDAExecutionProvider'],
808
+ provider_options=[{"device_id": device_id,
809
+ 'gpu_mem_limit': 2 * 1024 * 1024 * 1024}])
810
  self.face_app.prepare(ctx_id=device_id, det_size=(512, 512))
811
 
812
  self.device = device