Update app.py

app.py

@@ -31,7 +31,14 @@ noise_scheduler = DDIMScheduler(
     steps_offset=1,
 )
 vae = AutoencoderKL.from_pretrained(vae_model_path).to(dtype=torch.float16)
-pipe = pipeline.to(device)
+pipe = StableDiffusionPipeline.from_pretrained(
+    pipeline,
+    torch_dtype=torch.float16,
+    scheduler=noise_scheduler,
+    vae=vae,
+    feature_extractor=safety_feature_extractor,
+    safety_checker=safety_checker
+).to(device)
 
 ip_model = IPAdapterFaceID(pipe, ip_ckpt, device)
 ip_model_plus = IPAdapterFaceIDPlus(pipe, image_encoder_path, ip_plus_ckpt, device)