Spaces:

Suchinthana
/

VisualizeGeoMap

Runtime error

Suchinthana commited on Jan 22

Commit

0c6a8e1

1 Parent(s): 1b5be08

removing cpu offlload

Files changed (1) hide show

app.py CHANGED Viewed

@@ -110,7 +110,7 @@ controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_inpai
 pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
     "stable-diffusion-v1-5/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
 )
-pipeline.enable_model_cpu_offload()
 def make_inpaint_condition(init_image, mask_image):
     init_image = np.array(init_image.convert("RGB")).astype(np.float32) / 255.0

 pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
     "stable-diffusion-v1-5/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
 )
+#pipeline.enable_model_cpu_offload()
 def make_inpaint_condition(init_image, mask_image):
     init_image = np.array(init_image.convert("RGB")).astype(np.float32) / 255.0