multimodalart HF staff commited on
Commit
247fbd7
β€’
1 Parent(s): d3a2b44

Update pipeline_stable_diffusion_xl_instantid_img2img.py

Browse files
pipeline_stable_diffusion_xl_instantid_img2img.py CHANGED
@@ -534,8 +534,8 @@ class StableDiffusionXLInstantIDImg2ImgPipeline(StableDiffusionXLControlNetImg2I
534
  prompt_image_emb = torch.cat([torch.zeros_like(prompt_image_emb), prompt_image_emb], dim=0)
535
  else:
536
  prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
537
-
538
- prompt_image_emb = self.image_proj_model(prompt_image_emb)
539
  return prompt_image_emb
540
 
541
  @torch.no_grad()
 
534
  prompt_image_emb = torch.cat([torch.zeros_like(prompt_image_emb), prompt_image_emb], dim=0)
535
  else:
536
  prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
537
+ image_proj_model_device = self.image_proj_model.to(device)
538
+ prompt_image_emb = image_proj_model_device(prompt_image_emb)
539
  return prompt_image_emb
540
 
541
  @torch.no_grad()