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)
|