change schedulers the special jax way because of course that's how it needs to be done

app.py

@@ -35,7 +35,13 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         revision="flax",
         dtype=jnp.bfloat16,
         )
-pipe.scheduler = FlaxDPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+scheduler, scheduler_state = FlaxDPMSolverMultistepScheduler.from_pretrained(
+    "./models/wd-1-5-b2-flax",
+    subfolder="scheduler"
+)
+params["scheduler"] = scheduler_state
+
+#scheduler = FlaxDPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()
 #pipe.enable_xformers_memory_efficient_attention()