Update app.py

app.py

@@ -115,8 +115,8 @@ function custom(){
 
 image_pipe = StableDiffusionPipeline.from_pretrained(base, torch_dtype=dtype, safety_checker=None).to(device)
 
-adapter = MotionAdapter.from_pretrained(motion_adapter, torch_dtype=dtype, safety_checker=None)
-vae = AutoencoderKL.from_pretrained(vae_id, torch_dtype=torch.float16)
+adapter = MotionAdapter.from_pretrained(motion_adapter, torch_dtype=dtype, safety_checker=None).to(device)
+vae = AutoencoderKL.from_pretrained(vae_id, torch_dtype=torch.float16).to(device)
 
 scheduler = DPMSolverMultistepScheduler.from_pretrained(
     base,
@@ -138,7 +138,7 @@ pipe = AnimateDiffPipeline.from_pretrained(
     base,
     motion_adapter=motion_adapter,
     vae=vae,
-    scheduler=scheduler,
+    scheduler=scheduler2,
     torch_dtype=torch.float16
 ).to(device)