switch to DPM

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import jax.numpy as jnp
 from diffusers import FlaxStableDiffusionControlNetPipeline, FlaxControlNetModel
-from diffusers import FlaxScoreSdeVeScheduler
+from diffusers import FlaxScoreSdeVeScheduler, FlaxDPMSolverMultistepScheduler
 import torch
 torch.backends.cuda.matmul.allow_tf32 = True
 import torchvision
@@ -35,7 +35,7 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         revision="flax",
         dtype=jnp.bfloat16,
         )
-pipe.scheduler = FlaxScoreSdeVeScheduler.from_config(pipe.scheduler.config)
+pipe.scheduler = FlaxDPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()
 #pipe.enable_xformers_memory_efficient_attention()