Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -1620,8 +1620,8 @@ if __name__ == "__main__":
|
|
| 1620 |
|
| 1621 |
args = parser.parse_args()
|
| 1622 |
|
| 1623 |
-
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 1624 |
-
pipe = AccDiffusionSDXLPipeline.from_pretrained(args.model_ckpt,
|
| 1625 |
|
| 1626 |
|
| 1627 |
# GRADIO MODE
|
|
|
|
| 1620 |
|
| 1621 |
args = parser.parse_args()
|
| 1622 |
|
| 1623 |
+
# vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
| 1624 |
+
pipe = AccDiffusionSDXLPipeline.from_pretrained(args.model_ckpt, torch_dtype=torch.float16).to("cuda")
|
| 1625 |
|
| 1626 |
|
| 1627 |
# GRADIO MODE
|