Update app.py

app.py

@@ -44,7 +44,6 @@ with open('loras.json', 'r') as f:
 #dtype = torch.bfloat16
 #base_model = "AlekseyCalvin/Artsy_Lite_Flux_v1_by_jurdn_Diffusers"
 #pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=dtype).to("cuda")
-##pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
 
 #pipe = diffusers.ZImagePipeline.from_pretrained("Disty0/Z-Image-Turbo-SDNQ-uint4-svd-r32", torch_dtype=torch.bfloat16)
 #torch.cuda.empty_cache()
@@ -52,7 +51,9 @@ pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF1
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
+#pipe.vae = AutoencoderTiny.from_pretrained("madebyollin/taef1", torch_dtype=torch.float16).to("cuda")
+pipe.vae = AutoencoderKL.from_pretrained("AlekseyCalvin/AnimeVAE_by_Anzhc_for_Flux_ZiT", torch_dtype=torch.float32).to("cuda")
+
 
 #pipe.vae = AutoencoderKL.from_pretrained("REPA-E/e2e-flux-vae", torch_dtype=torch.bfloat16).to("cuda")
 ##The repa-e vae generates extremely noisy outputs for some reason.