Spaces:
Running
on
Zero
Running
on
Zero
Suchinthana
commited on
Commit
·
4363790
1
Parent(s):
9690adb
Using inpaiting model2
Browse files
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("
|
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 |
)
|