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]
|