adaface-neurips commited on
Commit
578cfba
·
1 Parent(s): a1f4e87

Fix tiny bug

Browse files
Files changed (1) hide show
  1. adaface/face_id_to_ada_prompt.py +1 -1
adaface/face_id_to_ada_prompt.py CHANGED
@@ -811,7 +811,7 @@ class ConsistentID_ID2AdaPrompt(FaceID2AdaPrompt):
811
  print(f'On HF space. Arc2Face Face encoder already loaded on cpu.')
812
  return
813
 
814
- if str(device) == 'cpu' or self.is_on_hf_space:
815
  self.face_app = FaceAnalysis(name='buffalo_l', root='models/insightface',
816
  providers=['CPUExecutionProvider'])
817
  self.face_app.prepare(ctx_id=0, det_size=(512, 512))
 
811
  print(f'On HF space. Arc2Face Face encoder already loaded on cpu.')
812
  return
813
 
814
+ if str(device) == 'cpu':
815
  self.face_app = FaceAnalysis(name='buffalo_l', root='models/insightface',
816
  providers=['CPUExecutionProvider'])
817
  self.face_app.prepare(ctx_id=0, det_size=(512, 512))