Spaces:
Runtime error
Runtime error
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
|