Update pipeline_stable_diffusion_3_ipa.py
Browse files
pipeline_stable_diffusion_3_ipa.py
CHANGED
@@ -1224,7 +1224,7 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
|
|
1224 |
print('shape 3.1: ', zeros_tensora.shape)
|
1225 |
clip_image_embedsa = average_image_embed.repeat(1, 1, 1)
|
1226 |
print('shape 3.5: ', clip_image_embedsa.shape)
|
1227 |
-
clip_image_embedse = torch.cat([zeros_tensora,
|
1228 |
print('shape 3.8: ', clip_image_embedse.shape)
|
1229 |
# 4. Concatenate the zeros and the average embedding
|
1230 |
clip_image_embeds = torch.cat([zeros_tensor, average_image_embed], dim=0)
|
|
|
1224 |
print('shape 3.1: ', zeros_tensora.shape)
|
1225 |
clip_image_embedsa = average_image_embed.repeat(1, 1, 1)
|
1226 |
print('shape 3.5: ', clip_image_embedsa.shape)
|
1227 |
+
clip_image_embedse = torch.cat([zeros_tensora, clip_image_embedsa], dim=0)
|
1228 |
print('shape 3.8: ', clip_image_embedse.shape)
|
1229 |
# 4. Concatenate the zeros and the average embedding
|
1230 |
clip_image_embeds = torch.cat([zeros_tensor, average_image_embed], dim=0)
|