Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ DEFAULT_WIDTH = 1024
|
|
| 27 |
|
| 28 |
REPO = "sd-community/sdxl-flash"
|
| 29 |
|
| 30 |
-
|
| 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)
|