Spaces:
Paused
Paused
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 =
|
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()
|