Hugging Face's logo

Spaces:
kevinppaulo
/
PuLID
Running on Zero

App Files Community
yanze commited on
Commit
61212bc
•
1 Parent(s): 3d16536

Update pulid/pipeline_flux.py

Browse files
Files changed (1) hide show
  1. pulid/pipeline_flux.py +1 -0
pulid/pipeline_flux.py CHANGED
@@ -138,6 +138,7 @@ class PuLIDPipeline(nn.Module):
138
  self.face_helper.read_image(image_bgr)
139
  print('face_det_device: ', self.face_helper.face_det.device)
140
  print('face_det_mean_tensor_device: ', self.face_helper.face_det.mean_tensor.device)
 
141
  self.face_helper.get_face_landmarks_5(only_center_face=True)
142
  self.face_helper.align_warp_face()
143
  if len(self.face_helper.cropped_faces) == 0:
 
138
  self.face_helper.read_image(image_bgr)
139
  print('face_det_device: ', self.face_helper.face_det.device)
140
  print('face_det_mean_tensor_device: ', self.face_helper.face_det.mean_tensor.device)
141
+ self.face_helper.face_det.mean_tensor.to(self.device)
142
  self.face_helper.get_face_landmarks_5(only_center_face=True)
143
  self.face_helper.align_warp_face()
144
  if len(self.face_helper.cropped_faces) == 0: