Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, model=D
|
|
55 |
if model == "Anime":
|
56 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
57 |
controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
|
58 |
-
repo = StableDiffusionXLPipeline.from_pretrained(cagliostrolab/animagine-xl-3.1, vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
59 |
steps = steps or 16
|
60 |
guidance = guidance or 7
|
61 |
else:
|
|
|
55 |
if model == "Anime":
|
56 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
57 |
controlnet = ControlNetModel.from_pretrained("MakiPan/controlnet-encoded-hands-130k", torch_dtype=torch.float16)
|
58 |
+
repo = StableDiffusionXLPipeline.from_pretrained("cagliostrolab/animagine-xl-3.1", vae=vae, controlnet=controlnet, torch_dtype=torch.float16, use_safetensors=True, add_watermarker=False)
|
59 |
steps = steps or 16
|
60 |
guidance = guidance or 7
|
61 |
else:
|