Spaces:
Runtime error
Runtime error
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.
|
| 43 |
-
dit.
|
| 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('
|
| 88 |
-
ckpt_path = 'models/
|
| 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)
|