BertChristiaens commited on
Commit
d47ae93
1 Parent(s): 3d4d894

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +1 -1
models.py CHANGED
@@ -82,7 +82,7 @@ def get_controlnet() -> ControlNetModel:
82
  ControlNetModel: controlnet model
83
  """
84
  controlnet = ControlNetModel.from_pretrained(
85
- "lllyasviel/control_v11p_sd15_seg", torch_dtype=torch.float16)
86
 
87
  pipe = StableDiffusionControlNetInpaintImg2ImgPipeline.from_pretrained(
88
  "runwayml/stable-diffusion-inpainting",
 
82
  ControlNetModel: controlnet model
83
  """
84
  controlnet = ControlNetModel.from_pretrained(
85
+ "BertChristiaens/controlnet-seg-room", torch_dtype=torch.float16)
86
 
87
  pipe = StableDiffusionControlNetInpaintImg2ImgPipeline.from_pretrained(
88
  "runwayml/stable-diffusion-inpainting",