SunderAli17 commited on
Commit
52f9ca7
1 Parent(s): 66b3640

Update eva_clip/model_configs/fluxpipeline.py

Browse files
eva_clip/model_configs/fluxpipeline.py CHANGED
@@ -39,8 +39,8 @@ class ToonMagePipeline(nn.Module):
39
  ])
40
 
41
  dit.pulid_ca = self.pulid_ca
42
- dit.toonmage_double_interval = double_interval
43
- dit.toonmage_single_interval = single_interval
44
 
45
  # preprocessors
46
  # face align and parsing
@@ -84,8 +84,8 @@ class ToonMagePipeline(nn.Module):
84
  self.debug_img_list = []
85
 
86
  def load_pretrain(self, pretrain_path=None):
87
- hf_hub_download('SunderAli17/SAK', 'toonmage_flux_v2.safetensors', local_dir ='models')
88
- ckpt_path = 'models/toonmage_flux_v2.safetensors'
89
  if pretrain_path is not None:
90
  ckpt_path = pretrain_path
91
  state_dict = load_file(ckpt_path)
 
39
  ])
40
 
41
  dit.pulid_ca = self.pulid_ca
42
+ dit.pulid_double_interval = double_interval
43
+ dit.pulid_single_interval = single_interval
44
 
45
  # preprocessors
46
  # face align and parsing
 
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)