face-to-all

Running on Zero

multimodalart HF staff commited on Apr 2

Commit

247fbd7

•

1 Parent(s): d3a2b44

Update pipeline_stable_diffusion_xl_instantid_img2img.py

Files changed (1) hide show

pipeline_stable_diffusion_xl_instantid_img2img.py CHANGED Viewed

@@ -534,8 +534,8 @@ class StableDiffusionXLInstantIDImg2ImgPipeline(StableDiffusionXLControlNetImg2I
             prompt_image_emb = torch.cat([torch.zeros_like(prompt_image_emb), prompt_image_emb], dim=0)
         else:
             prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
-        prompt_image_emb = self.image_proj_model(prompt_image_emb)
         return prompt_image_emb
     @torch.no_grad()

             prompt_image_emb = torch.cat([torch.zeros_like(prompt_image_emb), prompt_image_emb], dim=0)
         else:
             prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
+        image_proj_model_device = self.image_proj_model.to(device)
+        prompt_image_emb = image_proj_model_device(prompt_image_emb)
         return prompt_image_emb
     @torch.no_grad()