Update pipeline_stable_diffusion_3_ipa.py
Browse files
pipeline_stable_diffusion_3_ipa.py
CHANGED
@@ -1188,9 +1188,9 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
|
|
1188 |
# FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
|
1189 |
|
1190 |
clip_image_embeds = torch.mean(torch.stack(image_prompt_embeds_list), dim=0).unsqueeze(0)
|
1191 |
-
bs_embed, seq_len, _ =
|
1192 |
-
|
1193 |
-
|
1194 |
|
1195 |
#clip_image_embeds = torch.cat([torch.stack(image_prompt_embeds_list)], dim=0).mean(dim=0)
|
1196 |
|
|
|
1188 |
# FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
|
1189 |
|
1190 |
clip_image_embeds = torch.mean(torch.stack(image_prompt_embeds_list), dim=0).unsqueeze(0)
|
1191 |
+
bs_embed, seq_len, _ = clip_image_embeds.shape
|
1192 |
+
clip_image_embeds = clip_image_embeds.repeat(1, 1, 1)
|
1193 |
+
clip_image_embeds = clip_image_embeds.view(bs_embed * 1, seq_len, -1)
|
1194 |
|
1195 |
#clip_image_embeds = torch.cat([torch.stack(image_prompt_embeds_list)], dim=0).mean(dim=0)
|
1196 |
|