1inkusFace commited on
Commit
c3e9db2
·
verified ·
1 Parent(s): 2d5a18b

Update pipeline_stable_diffusion_3_ipa.py

Browse files
Files changed (1) hide show
  1. pipeline_stable_diffusion_3_ipa.py +2 -0
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: