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