controlnet-interior-design

Running on A10G

BertChristiaens commited on May 2, 2023

Commit

d47ae93

1 Parent(s): 3d4d894

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -82,7 +82,7 @@ def get_controlnet() -> ControlNetModel:
         ControlNetModel: controlnet model
     """
     controlnet = ControlNetModel.from_pretrained(
-        "lllyasviel/control_v11p_sd15_seg", torch_dtype=torch.float16)
     pipe = StableDiffusionControlNetInpaintImg2ImgPipeline.from_pretrained(
         "runwayml/stable-diffusion-inpainting",

         ControlNetModel: controlnet model
     """
     controlnet = ControlNetModel.from_pretrained(
+        "BertChristiaens/controlnet-seg-room", torch_dtype=torch.float16)
     pipe = StableDiffusionControlNetInpaintImg2ImgPipeline.from_pretrained(
         "runwayml/stable-diffusion-inpainting",