Hugging Face's logo

Spaces:
Statical
/
Image
Running on Zero

App Files Community
Staticaliza commited on
Commit
ba8b16b
·
verified ·
1 Parent(s): 44745aa

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
- AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
31
  controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float32)
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")
31
  controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float32)
32
 
33
  model = StableDiffusionXLPipeline.from_pretrained(REPO, vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)