Update app.py

app.py

@@ -13,9 +13,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 #good_vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae", torch_dtype=dtype).to(device)
 
 # from bobber/bigasp2 to John6666/biglove-ponyv20-sdxl
-# good_vae = AutoencoderKL.from_pretrained("John6666/biglove-ponyv20-sdxl", subfolder="vae", torch_dtype=dtype).to(device)
-# pipeline = StableDiffusionXLPipeline.from_pretrained("John6666/biglove-ponyv20-sdxl", torch_dtype=dtype, vae=good_vae).to(device)
-pipeline = StableDiffusionXLPipeline.from_pretrained("John6666/biglove-ponyv20-sdxl", torch_dtype=dtype).to(device)
+good_vae = AutoencoderKL.from_pretrained("bobber/bigasp2", subfolder="vae", torch_dtype=dtype).to(device)
+pipeline = StableDiffusionXLPipeline.from_pretrained("John6666/biglove-ponyv20-sdxl", torch_dtype=dtype, vae=good_vae).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max