yanze commited on
Commit
40f8743
1 Parent(s): 03aac7c

Update pulid/pipeline_flux.py

Browse files
Files changed (1) hide show
  1. pulid/pipeline_flux.py +2 -2
pulid/pipeline_flux.py CHANGED
@@ -84,8 +84,8 @@ class PuLIDPipeline(nn.Module):
84
  self.debug_img_list = []
85
 
86
  def load_pretrain(self, pretrain_path=None):
87
- hf_hub_download('guozinan/PuLID', 'pulid_flux_v0.9.0.safetensors', local_dir='models')
88
- ckpt_path = 'models/pulid_flux_v0.9.0.safetensors'
89
  if pretrain_path is not None:
90
  ckpt_path = pretrain_path
91
  state_dict = load_file(ckpt_path)
 
84
  self.debug_img_list = []
85
 
86
  def load_pretrain(self, pretrain_path=None):
87
+ hf_hub_download('guozinan/PuLID', 'pulid_flux_v0.9.1.safetensors', local_dir='models')
88
+ ckpt_path = 'models/pulid_flux_v0.9.1.safetensors'
89
  if pretrain_path is not None:
90
  ckpt_path = pretrain_path
91
  state_dict = load_file(ckpt_path)