Update inference_manager.py
Browse files- inference_manager.py +1 -1
inference_manager.py
CHANGED
@@ -501,7 +501,7 @@ class ModelManager:
|
|
501 |
start = time.time()
|
502 |
model.base_model_pipeline.to("cuda")
|
503 |
if not self.app:
|
504 |
-
self.app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider'
|
505 |
self.app.prepare(ctx_id=0, det_size=(512, 512))
|
506 |
print("extracting face...")
|
507 |
faceid_all_embeds = []
|
|
|
501 |
start = time.time()
|
502 |
model.base_model_pipeline.to("cuda")
|
503 |
if not self.app:
|
504 |
+
self.app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider'])#, 'CPUExecutionProvider'
|
505 |
self.app.prepare(ctx_id=0, det_size=(512, 512))
|
506 |
print("extracting face...")
|
507 |
faceid_all_embeds = []
|