Hugging Face's logo

Spaces:
Statical
/
Image
Running on Zero

App Files Community
Staticaliza commited on
Commit
9955db2
·
verified ·
1 Parent(s): 406a75d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ DEFAULT_WIDTH = 1024
27
 
28
  REPO = "sd-community/sdxl-flash"
29
 
30
- vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").half().to(DEVICE)
31
  controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
32
 
33
  model = StableDiffusionXLPipeline.from_pretrained(REPO, vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
 
27
 
28
  REPO = "sd-community/sdxl-flash"
29
 
30
+ vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse-original").half().to(DEVICE)
31
  controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
32
 
33
  model = StableDiffusionXLPipeline.from_pretrained(REPO, vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)