Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
20fe0e9
1
Parent(s):
a125540
Fix onnx option again
Browse files
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,
|
|
|
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,
|
|
|
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
|