Spaces:
Running
on
Zero
Running
on
Zero
Suchinthana
commited on
Commit
·
0c6a8e1
1
Parent(s):
1b5be08
removing cpu offlload
Browse files
app.py
CHANGED
|
@@ -110,7 +110,7 @@ controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_inpai
|
|
| 110 |
pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
| 111 |
"stable-diffusion-v1-5/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
|
| 112 |
)
|
| 113 |
-
pipeline.enable_model_cpu_offload()
|
| 114 |
|
| 115 |
def make_inpaint_condition(init_image, mask_image):
|
| 116 |
init_image = np.array(init_image.convert("RGB")).astype(np.float32) / 255.0
|
|
|
|
| 110 |
pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
|
| 111 |
"stable-diffusion-v1-5/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
|
| 112 |
)
|
| 113 |
+
#pipeline.enable_model_cpu_offload()
|
| 114 |
|
| 115 |
def make_inpaint_condition(init_image, mask_image):
|
| 116 |
init_image = np.array(init_image.convert("RGB")).astype(np.float32) / 255.0
|