nsfwalex commited on
Commit
65a5401
·
verified ·
1 Parent(s): d67a533

Update inference_manager.py

Browse files
Files changed (1) hide show
  1. 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', 'CPUExecutionProvider'])
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 = []