Update pipeline_stable_diffusion_3_ipa.py
Browse files
pipeline_stable_diffusion_3_ipa.py
CHANGED
@@ -1208,7 +1208,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
|
|
1208 |
clip_image_embeds_stack_list = torch.stack(image_prompt_embeds_list).mean(dim=0)
|
1209 |
print('stacked with mean dim 0 shape: ', clip_image_embeds_stack_list.shape)
|
1210 |
zeroes_tensor = torch.zeros_like(clip_image_embeds_stack_list)
|
1211 |
-
print('zeros shape: ',
|
1212 |
clip_image_embeds = torch.cat([zeroes_tensor, clip_image_embeds_stack_list], dim=0)
|
1213 |
print('embeds shape: ', clip_image_embeds.shape)
|
1214 |
|
|
|
1208 |
clip_image_embeds_stack_list = torch.stack(image_prompt_embeds_list).mean(dim=0)
|
1209 |
print('stacked with mean dim 0 shape: ', clip_image_embeds_stack_list.shape)
|
1210 |
zeroes_tensor = torch.zeros_like(clip_image_embeds_stack_list)
|
1211 |
+
print('zeros shape: ', zeroes_tensor.shape)
|
1212 |
clip_image_embeds = torch.cat([zeroes_tensor, clip_image_embeds_stack_list], dim=0)
|
1213 |
print('embeds shape: ', clip_image_embeds.shape)
|
1214 |
|