Spaces:
Runtime error
Runtime error
Update src/pipelines/instantid_single_pieline.py
Browse files
src/pipelines/instantid_single_pieline.py
CHANGED
@@ -234,6 +234,8 @@ class InstantidSingleConceptPipeline(StableDiffusionXLControlNetPipeline):
|
|
234 |
else:
|
235 |
prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
|
236 |
|
|
|
|
|
237 |
prompt_image_emb = self.image_proj_model(prompt_image_emb)
|
238 |
|
239 |
bs_embed, seq_len, _ = prompt_image_emb.shape
|
|
|
234 |
else:
|
235 |
prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
|
236 |
|
237 |
+
print(next(self.image_proj_model.parameters()).device)
|
238 |
+
print(prompt_image_emb.device)
|
239 |
prompt_image_emb = self.image_proj_model(prompt_image_emb)
|
240 |
|
241 |
bs_embed, seq_len, _ = prompt_image_emb.shape
|