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,
|