Spaces:
Running on Zero
Running on Zero
Upload 2 files
Browse files
pipeline_stable_diffusion_xl_instantid.py
CHANGED
|
@@ -540,6 +540,7 @@ class StableDiffusionXLInstantIDPipeline(StableDiffusionXLControlNetPipeline):
|
|
| 540 |
else:
|
| 541 |
prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
|
| 542 |
|
|
|
|
| 543 |
prompt_image_emb = self.image_proj_model(prompt_image_emb)
|
| 544 |
return prompt_image_emb
|
| 545 |
|
|
|
|
| 540 |
else:
|
| 541 |
prompt_image_emb = torch.cat([prompt_image_emb], dim=0)
|
| 542 |
|
| 543 |
+
self.image_proj_model.to(device=device, dtype=dtype)
|
| 544 |
prompt_image_emb = self.image_proj_model(prompt_image_emb)
|
| 545 |
return prompt_image_emb
|
| 546 |
|
requirements.txt
CHANGED
|
@@ -8,3 +8,4 @@ insightface
|
|
| 8 |
onnxruntime
|
| 9 |
opencv-python
|
| 10 |
huggingface-hub
|
|
|
|
|
|
| 8 |
onnxruntime
|
| 9 |
opencv-python
|
| 10 |
huggingface-hub
|
| 11 |
+
peft
|