linoyts HF staff commited on
Commit
6cf4fca
1 Parent(s): d97f88a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1718,9 +1718,10 @@ class LEditsPPPipelineStableDiffusionXL(
1718
 
1719
  #self.init_latents = xts[-1]
1720
  zs = zs.flip(0)
1721
-
1722
  if num_zero_noise_steps > 0:
1723
  zs[-num_zero_noise_steps:] = torch.zeros_like(zs[-num_zero_noise_steps:])
 
1724
  #self.zs = zs
1725
  #return LEditsPPInversionPipelineOutput(images=resized, vae_reconstruction_images=image_rec)
1726
  return xts[-1], zs
@@ -2043,6 +2044,7 @@ def invert_image(image, num_inversion_steps=50, skip=0.3):
2043
  num_inversion_steps = num_inversion_steps,
2044
  skip = skip
2045
  )
 
2046
  return init_latents,zs
2047
 
2048
  def reset_do_inversion():
 
1718
 
1719
  #self.init_latents = xts[-1]
1720
  zs = zs.flip(0)
1721
+ print("post 3")
1722
  if num_zero_noise_steps > 0:
1723
  zs[-num_zero_noise_steps:] = torch.zeros_like(zs[-num_zero_noise_steps:])
1724
+ print("post 4")
1725
  #self.zs = zs
1726
  #return LEditsPPInversionPipelineOutput(images=resized, vae_reconstruction_images=image_rec)
1727
  return xts[-1], zs
 
2044
  num_inversion_steps = num_inversion_steps,
2045
  skip = skip
2046
  )
2047
+ print("post 5")
2048
  return init_latents,zs
2049
 
2050
  def reset_do_inversion():