Spaces:
Runtime error
Runtime error
use TheMistoAI/MistoLine
Browse files
app.py
CHANGED
@@ -33,8 +33,14 @@ model = "stabilityai/stable-diffusion-xl-base-1.0"
|
|
33 |
# model = "stabilityai/sdxl-turbo"
|
34 |
# vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=dtype)
|
35 |
scheduler = DDIMScheduler.from_pretrained(model, subfolder="scheduler")
|
|
|
|
|
|
|
36 |
controlnet = ControlNetModel.from_pretrained(
|
37 |
-
"
|
|
|
|
|
|
|
38 |
)
|
39 |
pipe = StableDiffusionXLControlNetImg2ImgPipeline.from_pretrained(
|
40 |
model,
|
|
|
33 |
# model = "stabilityai/sdxl-turbo"
|
34 |
# vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=dtype)
|
35 |
scheduler = DDIMScheduler.from_pretrained(model, subfolder="scheduler")
|
36 |
+
# controlnet = ControlNetModel.from_pretrained(
|
37 |
+
# "diffusers/controlnet-canny-sdxl-1.0", torch_dtype=torch.float16
|
38 |
+
# )
|
39 |
controlnet = ControlNetModel.from_pretrained(
|
40 |
+
"TheMistoAI/MistoLine",
|
41 |
+
torch_dtype=torch.float16,
|
42 |
+
revision="refs/pr/3",
|
43 |
+
variant="fp16",
|
44 |
)
|
45 |
pipe = StableDiffusionXLControlNetImg2ImgPipeline.from_pretrained(
|
46 |
model,
|