thought I changed that

app.py

@@ -35,7 +35,7 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         revision="flax",
         dtype=jnp.bfloat16,
         )
-pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
+pipe.scheduler = FlaxScoreSdeVeScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()
 #pipe.enable_xformers_memory_efficient_attention()