Suchinthana commited on
Commit
4363790
·
1 Parent(s): 9690adb

Using inpaiting model2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -204,7 +204,7 @@ def generate_static_map(geojson_data, invisible=False):
204
  return m.render()
205
 
206
  # ControlNet pipeline setup
207
- controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_inpaint", torch_dtype=torch.float16)
208
  pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
209
  "stable-diffusion-v1-5/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
210
  )
 
204
  return m.render()
205
 
206
  # ControlNet pipeline setup
207
+ controlnet = ControlNetModel.from_pretrained("stabilityai/stable-diffusion-2-inpainting", torch_dtype=torch.float16)
208
  pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
209
  "stable-diffusion-v1-5/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
210
  )