use SDE solver

app.py

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