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

Update src/pipelines/instantid_single_pieline.py

Browse files
src/pipelines/instantid_single_pieline.py CHANGED
@@ -234,8 +234,8 @@ class InstantidSingleConceptPipeline(StableDiffusionXLControlNetPipeline):
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
 
234
  else:
235
  prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
236
 
237
+
238
+ self.image_proj_model.cuda()
239
  prompt_image_emb = self.image_proj_model(prompt_image_emb)
240
 
241
  bs_embed, seq_len, _ = prompt_image_emb.shape