Update app.py

app.py

@@ -52,6 +52,10 @@ pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF1
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe.vae = DiffusersAutoencoderKL.from_pretrained("kaiyuyue/FLUX.2-dev-vae", torch_dtype=torch.float16).to("cuda")
+vae_config = vae.config
+
+vae_config.scaling_factor = 0.3611  # Example value
+vae_config.shift_factor = 0.1159  # Example value
 
 #pipe.enable_model_cpu_offload()