Spaces:

Suchinthana
/

VisualizeGeoMap

Runtime error

Suchinthana commited on Jan 22

Commit

1b5be08

1 Parent(s): 6b6be99

model update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -108,7 +108,7 @@ def save_map_image(geojson_data):
 # ControlNet pipeline setup
 controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_inpaint", torch_dtype=torch.float16)
 pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
-    "runwayml/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
 )
 pipeline.enable_model_cpu_offload()

 # ControlNet pipeline setup
 controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_inpaint", torch_dtype=torch.float16)
 pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
+    "stable-diffusion-v1-5/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
 )
 pipeline.enable_model_cpu_offload()