Update app.py

app.py

@@ -1,5 +1,5 @@
 from hidiffusion import apply_hidiffusion, remove_hidiffusion
-from diffusers import DiffusionPipeline, DDIMScheduler, AutoencoderKL
+from diffusers import StableDiffusionPipeline, DDIMScheduler, AutoencoderKL
 from transformers import CLIPFeatureExtractor
 from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
 
@@ -11,7 +11,7 @@ safety_checker=StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diff
 feature_extractor=CLIPFeatureExtractor.from_pretrained("openai/clip-vit-base-patch32")
 pretrain_model = "SG161222/Realistic_Vision_V5.1_noVAE"
 scheduler = DDIMScheduler.from_pretrained(pretrain_model, subfolder="scheduler")
-pipe = DiffusionPipeline.from_pretrained(pretrain_model, scheduler = scheduler, safety_checker=safety_checker, feature_extractor=feature_extractor, vae=vae, torch_dtype=torch.float16).to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained(pretrain_model, scheduler = scheduler, safety_checker=safety_checker, feature_extractor=feature_extractor, vae=vae, torch_dtype=torch.float16).to("cuda")
 
 # # Optional. enable_xformers_memory_efficient_attention can save memory usage and increase inference speed. enable_model_cpu_offload and enable_vae_tiling can save memory usage.
 #pipe.enable_model_cpu_offload()