Update app.py

app.py

@@ -61,11 +61,13 @@ def generate(
 
         if not use_vae:  
             pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
-
+            pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+            
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
             pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
-        
+            pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+            
         if use_lora:
             pipe.load_lora_weights(lora)
             pipe.fuse_lora(lora_scale=0.7)