Spaces:
Running
on
Zero
Running
on
Zero
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.
|
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)
|