multimodalart HF staff commited on
Commit
a1f66f7
1 Parent(s): aa5a24b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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,