Spaces:
Runtime error
Runtime error
re-enable UniPC
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
|
35 |
revision="flax",
|
36 |
dtype=jnp.bfloat16
|
37 |
)
|
38 |
-
|
39 |
#pipe.enable_model_cpu_offload()
|
40 |
#pipe.enable_xformers_memory_efficient_attention()
|
41 |
|
@@ -98,7 +98,7 @@ gr.Interface(
|
|
98 |
outputs=gr.Gallery().style(grid=[2], height="auto"),
|
99 |
title="Generate controlled outputs with Categorical Conditioning on Waifu Diffusion 1.5 beta 2.",
|
100 |
description="This Space uses image examples as style conditioning.",
|
101 |
-
examples=[["1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, watercolor, night, turtleneck", "
|
102 |
allow_flagging=False,
|
103 |
).launch(enable_queue=True)
|
104 |
|
|
|
35 |
revision="flax",
|
36 |
dtype=jnp.bfloat16
|
37 |
)
|
38 |
+
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
39 |
#pipe.enable_model_cpu_offload()
|
40 |
#pipe.enable_xformers_memory_efficient_attention()
|
41 |
|
|
|
98 |
outputs=gr.Gallery().style(grid=[2], height="auto"),
|
99 |
title="Generate controlled outputs with Categorical Conditioning on Waifu Diffusion 1.5 beta 2.",
|
100 |
description="This Space uses image examples as style conditioning.",
|
101 |
+
examples=[["1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, watercolor, night, turtleneck", "realistic, real life", "wikipe_cond_1.png"]],
|
102 |
allow_flagging=False,
|
103 |
).launch(enable_queue=True)
|
104 |
|