Update app.py
Browse files
app.py
CHANGED
|
@@ -269,7 +269,7 @@ def generate_images_30(prompt, neg_prompt_1, neg_prompt_2, neg_prompt_3, width,
|
|
| 269 |
).images
|
| 270 |
|
| 271 |
# 2. Manually decode with our float32 VAE
|
| 272 |
-
latents_fp32 =
|
| 273 |
latents_fp32 = 1 / pipe.vae.config.scaling_factor * latents_fp32
|
| 274 |
with torch.no_grad():
|
| 275 |
# This is our high-precision sRGB tensor in range [-1, 1]
|
|
|
|
| 269 |
).images
|
| 270 |
|
| 271 |
# 2. Manually decode with our float32 VAE
|
| 272 |
+
latents_fp32 = sd_image.to(torch.float32)
|
| 273 |
latents_fp32 = 1 / pipe.vae.config.scaling_factor * latents_fp32
|
| 274 |
with torch.no_grad():
|
| 275 |
# This is our high-precision sRGB tensor in range [-1, 1]
|