radames commited on
Commit
f9b4389
1 Parent(s): c3dd0e2

use TheMistoAI/MistoLine

Browse files
Files changed (1) hide show
  1. app.py +7 -1
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
- "diffusers/controlnet-canny-sdxl-1.0", torch_dtype=torch.float16
 
 
 
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,