Update app.py

app.py

@@ -31,7 +31,7 @@ print(f'torch version:{torch.__version__}')
 
 import torch.utils.checkpoint
 from pytorch_lightning import seed_everything
-from diffusers import AutoencoderKL, DDPMScheduler
+from diffusers import AutoencoderKL, DDIMScheduler
 from diffusers.utils import check_min_version
 from diffusers.utils.import_utils import is_xformers_available
 from transformers import CLIPTextModel, CLIPTokenizer, CLIPImageProcessor
@@ -77,7 +77,7 @@ snapshot_download(
 pretrained_model_path = 'preset/models/sd-turbo'
 seesr_model_path = 'preset/models/seesr'
 
-scheduler = DDPMScheduler.from_pretrained(pretrained_model_path, subfolder="scheduler")
+scheduler = DDIMScheduler.from_pretrained(pretrained_model_path, subfolder="scheduler")
 text_encoder = CLIPTextModel.from_pretrained(pretrained_model_path, subfolder="text_encoder")
 tokenizer = CLIPTokenizer.from_pretrained(pretrained_model_path, subfolder="tokenizer")
 vae = AutoencoderKL.from_pretrained(pretrained_model_path, subfolder="vae")