process image

cc819010 · patil-suraj · 7ac909d6 · cc819010
Commit cc819010 authored Jun 10, 2022 by patil-suraj
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

models/vision/latent_diffusion/modeling_latent_diffusion.py models/vision/latent_diffusion/modeling_latent_diffusion.py +1 -0

No files found.
--- a/models/vision/latent_diffusion/modeling_latent_diffusion.py
+++ b/models/vision/latent_diffusion/modeling_latent_diffusion.py
@@ -961,5 +961,6 @@ class LatentDiffusion(DiffusionPipeline):
        image = 1 /  0.18215 * image
        image = self.vqvae.decode(image)
        image = torch.clamp((image+1.0)/2.0, min=0.0, max=1.0)
+        image = 255. * image

        return image