fix DDIM

app.py

@@ -91,9 +91,9 @@ def load_model(model_id, lora_strength, use_controlnet=False, control_mode="edge
                         "sayakpaul/taesd-diffusers", torch_dtype=torch.float16, use_safetensors=True,
                     )
                 elif acceleration_mode == "DDIM":
-                    pipe = StableDiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch_dtype)
-                    pipe.scheduler = DDIMScheduler.from_config(
-                        pipe.scheduler.config, rescale_betas_zero_snr=True, timestep_spacing="trailing"
+                    scheduler = DDIMScheduler.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", subfolder="scheduler")
+                    pipe = StableDiffusionPipeline.from_pretrained(
+                        "stable-diffusion-v1-5/stable-diffusion-v1-5", scheduler=scheduler, torch_dtype=torch.float16
                     )
 
         if use_ip_adapter: