Fucius commited on
Commit
412ff69
1 Parent(s): 5995597

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