Spaces:
Configuration error
Configuration 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
|