Update app.py

app.py

@@ -55,6 +55,8 @@ pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF1
 custom_vae = AutoencoderKL.from_pretrained(
     "AlekseyCalvin/AnimeVAE_by_Anzhc_for_Flux_ZiT",
     torch_dtype=torch.float32,
+    ignore_mismatched_sizes=True,
+    low_cpu_mem_usage=False,
     device_map=None # Do not use device_map here
 )