Spaces:
Runtime error
Runtime error
disable unipc and enable DPM solver
Browse files
app.py
CHANGED
@@ -33,9 +33,10 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
|
|
33 |
"./models/wd-1-5-b2-flax",
|
34 |
controlnet=cnet,
|
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 |
|
|
|
33 |
"./models/wd-1-5-b2-flax",
|
34 |
controlnet=cnet,
|
35 |
revision="flax",
|
36 |
+
dtype=jnp.bfloat16,
|
37 |
+
scheduler="FlaxDPMSolverMultistepScheduler"
|
38 |
)
|
39 |
+
#pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
40 |
#pipe.enable_model_cpu_offload()
|
41 |
#pipe.enable_xformers_memory_efficient_attention()
|
42 |
|