just get it running

app.py

@@ -34,7 +34,6 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         controlnet=cnet,
         revision="flax",
         dtype=jnp.bfloat16,
-        scheduler="FlaxDPMSolverMultistepScheduler"
         )
 #pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()
@@ -73,7 +72,7 @@ def infer(prompt, negative_prompt, image):
         image=cond_img_in,
         params=p_params,
         prng_seed=rng,
-        num_inference_steps=20,
+        num_inference_steps=50,
         neg_prompt_ids=n_prompt_in,
         jit=True,
             ).images