Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1619,8 +1619,9 @@ if __name__ == "__main__":
|
|
1619 |
parser.add_argument('--experiment_name', default="AccDiffusion")
|
1620 |
|
1621 |
args = parser.parse_args()
|
1622 |
-
|
1623 |
-
|
|
|
1624 |
|
1625 |
|
1626 |
# GRADIO MODE
|
|
|
1619 |
parser.add_argument('--experiment_name', default="AccDiffusion")
|
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, vae=vae, torch_dtype=torch.float16).to("cuda")
|
1625 |
|
1626 |
|
1627 |
# GRADIO MODE
|