Suchinthana commited on
Commit
3e8f3e6
·
1 Parent(s): 0c6a8e1

Pipeline to cuda update

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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