Update pipeline_stable_diffusion_3_ipa.py
Browse files
pipeline_stable_diffusion_3_ipa.py
CHANGED
@@ -1205,7 +1205,8 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
|
|
1205 |
# FAILS clip_image_embeds = torch.cat(torch.stack(image_prompt_embeds_list), dim=0).mean(dim=0)
|
1206 |
# FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
|
1207 |
|
1208 |
-
|
|
|
1209 |
# 1. Stack the image embeddings
|
1210 |
stacked_image_embeds = torch.stack(image_prompt_embeds_list)
|
1211 |
print('shape 1: ', stacked_image_embeds.shape)
|
|
|
1205 |
# FAILS clip_image_embeds = torch.cat(torch.stack(image_prompt_embeds_list), dim=0).mean(dim=0)
|
1206 |
# FAILS TIMESTEPS clip_image_embeds = torch.cat(image_prompt_embeds_list, dim=0).mean(dim=0)
|
1207 |
|
1208 |
+
clip_image_embedsl = self.encode_clip_image_emb(clip_image, device, dtype)
|
1209 |
+
print('old method shape: ', clip_image_embedsl.shape)
|
1210 |
# 1. Stack the image embeddings
|
1211 |
stacked_image_embeds = torch.stack(image_prompt_embeds_list)
|
1212 |
print('shape 1: ', stacked_image_embeds.shape)
|