disable unipc

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 = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 #pipe.enable_model_cpu_offload()
 #pipe.enable_xformers_memory_efficient_attention()
 
@@ -71,8 +71,8 @@ def infer(prompt, negative_prompt, image):
         image=cond_img_in,
         prng_seed=rng,
         neg_prompt_ids=n_prompt_in,
-        #num_inference_steps=20,
         params=p_params,
+        num_inference_steps=20,
         jit=True
             ).images