Spaces:
Runtime error
Runtime error
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 |
-
|
238 |
-
|
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
|