remove custom scheduler

app.py

@@ -3,7 +3,6 @@ import numpy as np
 import torch
 import jax
 import jax.numpy as jnp
-from diffusers import StableDiffusionInpaintPipeline
 from flax.jax_utils import replicate
 from flax.training.common_utils import shard
 from PIL import Image
@@ -41,9 +40,6 @@ pipe, params = FlaxStableDiffusionControlNetPipeline.from_pretrained(
 params["controlnet"] = controlnet_params
 p_params = replicate(params)
 
-pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
-pipe = pipe.to(device)
-
 
 with gr.Blocks() as demo:
     gr.Markdown("# WildSynth: Synthetic Wildlife Data Generation")