yanze commited on
Commit
0f90f5b
1 Parent(s): 57ef07b

Update pulid/pipeline_flux.py

Browse files
Files changed (1) hide show
  1. pulid/pipeline_flux.py +2 -1
pulid/pipeline_flux.py CHANGED
@@ -55,7 +55,8 @@ class PuLIDPipeline(nn.Module):
55
  )
56
  self.face_helper.face_parse = None
57
  self.face_helper.face_parse = init_parsing_model(model_name='bisenet', device=self.device)
58
- self.face_helper = self.face_helper.to(self.device)
 
59
  # clip-vit backbone
60
  model, _, _ = create_model_and_transforms('EVA02-CLIP-L-14-336', 'eva_clip', force_custom_clip=True)
61
  model = model.visual
 
55
  )
56
  self.face_helper.face_parse = None
57
  self.face_helper.face_parse = init_parsing_model(model_name='bisenet', device=self.device)
58
+ self.face_helper.face_parse = self.face_helper.face_parse.to(self.device)
59
+ self.face_helper.face_det = self.face_helper.face_det.to(self.device)
60
  # clip-vit backbone
61
  model, _, _ = create_model_and_transforms('EVA02-CLIP-L-14-336', 'eva_clip', force_custom_clip=True)
62
  model = model.visual