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