Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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():
|