Spaces:
Running
on
Zero
Running
on
Zero
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()
|