Spaces:
Build error
Build error
ResearcherXman
commited on
Commit
·
f43f8b1
1
Parent(s):
da665db
update
Browse files- app.py +0 -1
- pipeline_stable_diffusion_xl_instantid_full.py +2 -0
app.py
CHANGED
|
@@ -131,7 +131,6 @@ pipe.scheduler = diffusers.EulerDiscreteScheduler.from_config(
|
|
| 131 |
|
| 132 |
pipe.cuda()
|
| 133 |
pipe.load_ip_adapter_instantid(face_adapter)
|
| 134 |
-
print(pipe.image_proj_model.device, pipe.unet.device)
|
| 135 |
pipe.image_proj_model.to("cuda")
|
| 136 |
pipe.unet.to("cuda")
|
| 137 |
# load and disable LCM
|
|
|
|
| 131 |
|
| 132 |
pipe.cuda()
|
| 133 |
pipe.load_ip_adapter_instantid(face_adapter)
|
|
|
|
| 134 |
pipe.image_proj_model.to("cuda")
|
| 135 |
pipe.unet.to("cuda")
|
| 136 |
# load and disable LCM
|
pipeline_stable_diffusion_xl_instantid_full.py
CHANGED
|
@@ -613,6 +613,8 @@ class StableDiffusionXLInstantIDPipeline(StableDiffusionXLControlNetPipeline):
|
|
| 613 |
else:
|
| 614 |
prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
|
| 615 |
|
|
|
|
|
|
|
| 616 |
prompt_image_emb = self.image_proj_model(prompt_image_emb)
|
| 617 |
|
| 618 |
bs_embed, seq_len, _ = prompt_image_emb.shape
|
|
|
|
| 613 |
else:
|
| 614 |
prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
|
| 615 |
|
| 616 |
+
print(prompt_image_emb.device)
|
| 617 |
+
print(self.image_proj_model.device)
|
| 618 |
prompt_image_emb = self.image_proj_model(prompt_image_emb)
|
| 619 |
|
| 620 |
bs_embed, seq_len, _ = prompt_image_emb.shape
|