Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
a1f66f7
1
Parent(s):
aa5a24b
Update app.py
Browse files
app.py
CHANGED
@@ -4,6 +4,8 @@ import torch
|
|
4 |
from huggingface_hub import hf_hub_download
|
5 |
from safetensors.torch import load_file
|
6 |
|
|
|
|
|
7 |
### SDXL Turbo ####
|
8 |
pipe_turbo = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo",
|
9 |
vae=vae,
|
|
|
4 |
from huggingface_hub import hf_hub_download
|
5 |
from safetensors.torch import load_file
|
6 |
|
7 |
+
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
8 |
+
|
9 |
### SDXL Turbo ####
|
10 |
pipe_turbo = StableDiffusionXLPipeline.from_pretrained("stabilityai/sdxl-turbo",
|
11 |
vae=vae,
|