Hugging Face's logo

Spaces:
Statical
/
Image
Running on Zero

App Files Community
Staticaliza commited on
Commit
59877ea
·
verified ·
1 Parent(s): 268543e

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-original").half().to(DEVICE), vae=vae
31
  controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
32
 
33
  model = StableDiffusionXLPipeline.from_pretrained(REPO, 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("john-doe-12344/anything-v4.0-vae").half().to(DEVICE), vae=vae
31
  controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
32
 
33
  model = StableDiffusionXLPipeline.from_pretrained(REPO, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)