diff --git a/models/vision/latent_diffusion/modeling_latent_diffusion.py b/models/vision/latent_diffusion/modeling_latent_diffusion.py
index 99c9b2e053..cb83254409 100644
--- 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