Spaces:
Running
on
Zero
Running
on
Zero
Suchinthana
commited on
Commit
·
3e8f3e6
1
Parent(s):
0c6a8e1
Pipeline to cuda update
Browse files
app.py
CHANGED
@@ -111,6 +111,8 @@ 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
|
|
|
111 |
"stable-diffusion-v1-5/stable-diffusion-inpainting", controlnet=controlnet, torch_dtype=torch.float16
|
112 |
)
|
113 |
#pipeline.enable_model_cpu_offload()
|
114 |
+
pipeline.to('cuda')
|
115 |
+
|
116 |
|
117 |
def make_inpaint_condition(init_image, mask_image):
|
118 |
init_image = np.array(init_image.convert("RGB")).astype(np.float32) / 255.0
|