Skip to content

Commit 6761336

Browse files
kaixuanliuyiyixuxu
andauthored
fix: 'PaintByExampleImageEncoder' object has no attribute 'all_tied_w… (#13252)
* fix: 'PaintByExampleImageEncoder' object has no attribute 'all_tied_weights_keys' Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> * also fix LDMBertModel Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> --------- Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com> Co-authored-by: YiYi Xu <yixu310@gmail.com>
1 parent 0c01a4b commit 6761336

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -720,6 +720,7 @@ def __init__(self, config: LDMBertConfig):
720720
super().__init__(config)
721721
self.model = LDMBertEncoder(config)
722722
self.to_logits = nn.Linear(config.hidden_size, config.vocab_size)
723+
self.post_init()
723724

724725
def forward(
725726
self,

src/diffusers/pipelines/paint_by_example/image_encoder.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,8 @@ def __init__(self, config, proj_size=None):
3535
# uncondition for scaling
3636
self.uncond_vector = nn.Parameter(torch.randn((1, 1, self.proj_size)))
3737

38+
self.post_init()
39+
3840
def forward(self, pixel_values, return_uncond_vector=False):
3941
clip_output = self.model(pixel_values=pixel_values)
4042
latent_states = clip_output.pooler_output

0 commit comments

Comments
 (0)