Update inference_manager.py
Browse files- inference_manager.py +1 -0
inference_manager.py
CHANGED
@@ -502,6 +502,7 @@ class ModelManager:
|
|
502 |
raise Exception(f"face images not provided")
|
503 |
start = time.time()
|
504 |
ip_model.pipe.to("cuda")
|
|
|
505 |
if not self.app:
|
506 |
self.app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider'])#, 'CPUExecutionProvider'
|
507 |
self.app.prepare(ctx_id=0, det_size=(512, 512))
|
|
|
502 |
raise Exception(f"face images not provided")
|
503 |
start = time.time()
|
504 |
ip_model.pipe.to("cuda")
|
505 |
+
ip_model.image_proj_model.to("cuda")
|
506 |
if not self.app:
|
507 |
self.app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider'])#, 'CPUExecutionProvider'
|
508 |
self.app.prepare(ctx_id=0, det_size=(512, 512))
|