yanze commited on
Commit
fd227fe
1 Parent(s): 61212bc

Update pulid/pipeline_flux.py

Browse files
Files changed (1) hide show
  1. pulid/pipeline_flux.py +1 -1
pulid/pipeline_flux.py CHANGED
@@ -138,7 +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.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:
 
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 = 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: