xywwww commited on
Commit
4230de2
·
verified ·
1 Parent(s): 7eb2bf1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
17
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
18
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
19
  "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
20
- ).to("cuda")
21
 
22
  def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode, strength, scale, seed, eta, low_threshold, high_threshold):
23
  with torch.no_grad():
 
17
  controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
18
  pipe = StableDiffusionControlNetPipeline.from_pretrained(
19
  "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
20
+ )
21
 
22
  def process(input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode, strength, scale, seed, eta, low_threshold, high_threshold):
23
  with torch.no_grad():