Spaces:
Running
on
Zero
Running
on
Zero
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():
|