Update pipeline_stable_diffusion_3_ipa.py
Browse files
pipeline_stable_diffusion_3_ipa.py
CHANGED
@@ -1149,8 +1149,10 @@ class StableDiffusion3Pipeline(DiffusionPipeline, SD3LoraLoaderMixin, FromSingle
|
|
1149 |
clip_image = clip_image.resize((max(clip_image.size), max(clip_image.size)))
|
1150 |
#clip_image_embeds_1 = self.encode_clip_image_emb(clip_image, device, dtype)
|
1151 |
clip_image_embeds_1 = self.clip_image_processor(images=clip_image, return_tensors="pt").pixel_values
|
|
|
1152 |
clip_image_embeds_1 = clip_image_embeds_1.to(device, dtype=dtype)
|
1153 |
clip_image_embeds_1 = self.image_encoder(clip_image_embeds_1, output_hidden_states=True).hidden_states[-2]
|
|
|
1154 |
clip_image_embeds_1 = clip_image_embeds_1 * scale_1
|
1155 |
image_prompt_embeds_list.append(clip_image_embeds_1)
|
1156 |
if clip_image_2 != None:
|
|
|
1149 |
clip_image = clip_image.resize((max(clip_image.size), max(clip_image.size)))
|
1150 |
#clip_image_embeds_1 = self.encode_clip_image_emb(clip_image, device, dtype)
|
1151 |
clip_image_embeds_1 = self.clip_image_processor(images=clip_image, return_tensors="pt").pixel_values
|
1152 |
+
print('clip output size: ', clip_image_embeds_1.shape)
|
1153 |
clip_image_embeds_1 = clip_image_embeds_1.to(device, dtype=dtype)
|
1154 |
clip_image_embeds_1 = self.image_encoder(clip_image_embeds_1, output_hidden_states=True).hidden_states[-2]
|
1155 |
+
print('encoder output size: ', clip_image_embeds_1.shape)
|
1156 |
clip_image_embeds_1 = clip_image_embeds_1 * scale_1
|
1157 |
image_prompt_embeds_list.append(clip_image_embeds_1)
|
1158 |
if clip_image_2 != None:
|