Spaces:
Sleeping
Sleeping
upd
Browse files
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=
|
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]
|