process image

2022-06-10 01:12:23 +02:00 · 2022-06-10 01:12:23 +02:00 · cc81901054
parent 7ac909d62a
commit cc81901054
1 changed files with 1 additions and 0 deletions
--- 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