yisol
commited on
Commit
•
cfede52
1
Parent(s):
1a13129
update demo code
Browse files- src/tryon_pipeline.py +1 -1
src/tryon_pipeline.py
CHANGED
|
@@ -1720,7 +1720,7 @@ class StableDiffusionXLInpaintPipeline(
|
|
| 1720 |
)
|
| 1721 |
|
| 1722 |
#project outside for loop
|
| 1723 |
-
image_embeds = unet.encoder_hid_proj(image_embeds).to(prompt_embeds.dtype)
|
| 1724 |
|
| 1725 |
|
| 1726 |
# 11. Denoising loop
|
|
|
|
| 1720 |
)
|
| 1721 |
|
| 1722 |
#project outside for loop
|
| 1723 |
+
image_embeds = self.unet.encoder_hid_proj(image_embeds).to(prompt_embeds.dtype)
|
| 1724 |
|
| 1725 |
|
| 1726 |
# 11. Denoising loop
|