Update pipeline_stable_diffusion_3_ipa.py
Browse files
pipeline_stable_diffusion_3_ipa.py
CHANGED
@@ -1217,10 +1217,12 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
|
|
1217 |
print('shape 2a: ', average_image_embedf.shape)
|
1218 |
# 3. Create a tensor of zeros with the same shape as the averaged embedding
|
1219 |
zeros_tensor = torch.zeros_like(average_image_embed)
|
|
|
1220 |
print('shape 3: ', zeros_tensor.shape)
|
|
|
1221 |
# 4. Concatenate the zeros and the average embedding
|
1222 |
clip_image_embeds = torch.cat([zeros_tensor, average_image_embed], dim=0)
|
1223 |
-
clip_image_embeds2 = torch.cat([
|
1224 |
print('shape 4: ', clip_image_embeds.shape)
|
1225 |
print('shape 4a: ', clip_image_embeds2.shape)
|
1226 |
'''
|
|
|
1217 |
print('shape 2a: ', average_image_embedf.shape)
|
1218 |
# 3. Create a tensor of zeros with the same shape as the averaged embedding
|
1219 |
zeros_tensor = torch.zeros_like(average_image_embed)
|
1220 |
+
zeros_tensora = torch.zeros_like(average_image_embed)
|
1221 |
print('shape 3: ', zeros_tensor.shape)
|
1222 |
+
print('shape 3a: ', zeros_tensora.shape)
|
1223 |
# 4. Concatenate the zeros and the average embedding
|
1224 |
clip_image_embeds = torch.cat([zeros_tensor, average_image_embed], dim=0)
|
1225 |
+
clip_image_embeds2 = torch.cat([zeros_tensora, average_image_embedf], dim=0)
|
1226 |
print('shape 4: ', clip_image_embeds.shape)
|
1227 |
print('shape 4a: ', clip_image_embeds2.shape)
|
1228 |
'''
|