Spaces:

fffiloni
/

AccDiffusion

Running on Zero

fffiloni commited on Jul 31

Commit

6c4e820

•

1 Parent(s): 09cc71e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1620,8 +1620,8 @@ if __name__ == "__main__":
     args = parser.parse_args()
-    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

     args = parser.parse_args()
+    # vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
+    pipe = AccDiffusionSDXLPipeline.from_pretrained(args.model_ckpt, torch_dtype=torch.float16).to("cuda")
     # GRADIO MODE