nsfwalex commited on
Commit
73b863a
·
verified ·
1 Parent(s): 9555f5e

Update inference_manager.py

Browse files
Files changed (1) hide show
  1. 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))