yisol commited on
Commit
cfede52
1 Parent(s): 1a13129

update demo code

Browse files
Files changed (1) hide show
  1. 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