Update app.py

app.py

@@ -1619,8 +1619,9 @@ if __name__ == "__main__":
     parser.add_argument('--experiment_name', default="AccDiffusion")
 
     args = parser.parse_args()
-    
-    pipe = AccDiffusionSDXLPipeline.from_pretrained(args.model_ckpt, torch_dtype=torch.float16).to("cuda")
+
+    vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+    pipe = AccDiffusionSDXLPipeline.from_pretrained(args.model_ckpt, vae=vae, torch_dtype=torch.float16).to("cuda")
     
     
     # GRADIO MODE