Spaces:
Running
on
Zero
Running
on
Zero
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,
|