Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -142,31 +142,6 @@ def latents_to_pil(latents):
|
|
| 142 |
pil_images = [Image.fromarray(image) for image in images]
|
| 143 |
return pil_images
|
| 144 |
|
| 145 |
-
"""We'll use a pic from the web here, but you can load your own instead by uploading it and editing the filename in the next cell."""
|
| 146 |
-
|
| 147 |
-
|
| 148 |
-
# Setting the number of sampling steps:
|
| 149 |
-
set_timesteps(scheduler, 15)
|
| 150 |
-
|
| 151 |
-
"""You can see how our new set of steps corresponds to those used in training:"""
|
| 152 |
-
|
| 153 |
-
# See these in terms of the original 1000 steps used for training:
|
| 154 |
-
print(scheduler.timesteps)
|
| 155 |
-
|
| 156 |
-
"""And how much noise is present at each:"""
|
| 157 |
-
|
| 158 |
-
# Look at the equivalent noise levels:
|
| 159 |
-
print(scheduler.sigmas)
|
| 160 |
-
|
| 161 |
-
# TODO maybe show timestep as well
|
| 162 |
-
|
| 163 |
-
"""This 'sigma' is the amount of noise added to the latent representation. Let's visualize what this looks like by adding a bit of noise to our encoded image and then decoding this noised version:"""
|
| 164 |
-
|
| 165 |
-
noise = torch.randn_like(encoded) # Random noise
|
| 166 |
-
sampling_step = 10 # Equivalent to step 10 out of 15 in the schedule above
|
| 167 |
-
# encoded_and_noised = scheduler.add_noise(encoded, noise, timestep) # Diffusers 0.3 and below
|
| 168 |
-
encoded_and_noised = scheduler.add_noise(encoded, noise, timesteps=torch.tensor([scheduler.timesteps[sampling_step]]))
|
| 169 |
-
latents_to_pil(encoded_and_noised.float())[0] # Display
|
| 170 |
|
| 171 |
"""What does this look like at different timesteps? Experiment and see for yourself!
|
| 172 |
|
|
|
|
| 142 |
pil_images = [Image.fromarray(image) for image in images]
|
| 143 |
return pil_images
|
| 144 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 145 |
|
| 146 |
"""What does this look like at different timesteps? Experiment and see for yourself!
|
| 147 |
|