disabling for now

app.py

@@ -35,11 +35,11 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
         revision="flax",
         dtype=jnp.bfloat16,
         )
-scheduler, scheduler_state = FlaxDPMSolverMultistepScheduler.from_pretrained(
-    "./models/wd-1-5-b2-flax",
-    subfolder="scheduler"
-)
-params["scheduler"] = scheduler_state
+#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()
@@ -78,7 +78,7 @@ def infer(prompt, negative_prompt, image):
         image=cond_img_in,
         params=p_params,
         prng_seed=rng,
-        num_inference_steps=20,
+        num_inference_steps=70,
         neg_prompt_ids=n_prompt_in,
         jit=True,
             ).images