Spaces:
Running
on
Zero
Running
on
Zero
Update pulid/pipeline_flux.py
Browse files- 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.
|
88 |
-
ckpt_path = 'models/pulid_flux_v0.9.
|
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)
|