oshita-n commited on
Commit
92a1c24
·
1 Parent(s): 17a1c6f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,14 +39,14 @@ def process_image(input_image: Image.Image, to_grayscale: bool, prompt: str) ->
39
  pipe.enable_model_cpu_offload()
40
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
41
  pipe.enable_xformers_memory_efficient_attention()
42
-
43
  generator = torch.manual_seed(2)
44
  output = pipe(
45
  prompt,
46
  canny_output,
47
  negative_prompt="monochrome, lowres, bad anatomy, worst quality, low quality",
48
  generator=generator,
49
- num_inference_steps=2,
50
  )
51
 
52
  return output.images[0]
 
39
  pipe.enable_model_cpu_offload()
40
  pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
41
  pipe.enable_xformers_memory_efficient_attention()
42
+
43
  generator = torch.manual_seed(2)
44
  output = pipe(
45
  prompt,
46
  canny_output,
47
  negative_prompt="monochrome, lowres, bad anatomy, worst quality, low quality",
48
  generator=generator,
49
+ num_inference_steps=20,
50
  )
51
 
52
  return output.images[0]