Update app.py

app.py

@@ -52,8 +52,8 @@ def nms(x, t, s):
     return z
 
 
-DESCRIPTION = '''# Scribble SDXL 🖋️🌄
-sketch to image with SDXL, using [@xinsir](https://huggingface.co/xinsir) [scribble sdxl controlnet](https://huggingface.co/xinsir/controlnet-scribble-sdxl-1.0)
+DESCRIPTION = '''# Flash Scribble SDXL 🖋️🌄
+super fast sketch to image with Flash SDXL, using [@xinsir](https://huggingface.co/xinsir) [scribble sdxl controlnet](https://huggingface.co/xinsir/controlnet-scribble-sdxl-1.0) and [sdxl flash](https://huggingface.co/sd-community/sdxl-flash)
 '''
 
 if not torch.cuda.is_available():
@@ -124,9 +124,6 @@ def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-eulera_scheduler = EulerAncestralDiscreteScheduler.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", subfolder="scheduler")
-
-
 controlnet = ControlNetModel.from_pretrained(
     "xinsir/controlnet-scribble-sdxl-1.0",
     torch_dtype=torch.float16
@@ -136,12 +133,13 @@ controlnet = ControlNetModel.from_pretrained(
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 
 pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
+    "sd-community/sdxl-flash",
     controlnet=controlnet,
     vae=vae,
     torch_dtype=torch.float16,
-    scheduler=eulera_scheduler,
+    # scheduler=eulera_scheduler,
 )
+pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 pipe.to(device)
 # Load model.