Spaces:
Build error
Build error
Commit
·
a125540
1
Parent(s):
a9ea043
Fix onnx option bug
Browse files
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'
|
662 |
-
provider_options=[{"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'
|
807 |
-
provider_options=[{"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
|