Update app.py

app.py

@@ -37,8 +37,8 @@ sdxl_pipe.to(device)
 sdxl_flash_pipe = StableDiffusionXLPipeline.from_pretrained("sd-community/sdxl-flash", torch_dtype=torch.float16)
 sdxl_flash_pipe.to(device)
 
-# Ensure sampler uses "trailing" timesteps.
-sdxl_flash_pipe.scheduler = DPMSolverSinglestepScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
+# Ensure sampler uses "trailing" timesteps for sdxl flash.
+sdxl_flash_pipe.scheduler = DPMSolverSinglestepScheduler.from_config(sdxl_flash_pipe.scheduler.config, timestep_spacing="trailing")
 
 # Define the image generation function for the Arena tab
 @spaces.GPU(duration=80)