Update app.py

app.py

@@ -186,7 +186,7 @@ class FlowMatchingPipeline:
                 )[0]
         
         # Decode latents
-        latents = latents / self.vae_scale_factor
+        latents = (latents / self.vae_scale_factor) * 5.52
         
         with torch.no_grad():
             image = self.vae.decode(latents).sample