Update app.py

app.py

@@ -164,7 +164,7 @@ def main():
     global step
     global dtype
 
-    device = "cuda"
+    device = "auto"
     dtype = torch.float16
     result=[]
     step = 2
@@ -182,7 +182,7 @@ def main():
     adapter = MotionAdapter().to(device, dtype)
     adapter.load_state_dict(load_file(hf_hub_download(repo ,ckpt), device=device))
 
-    pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, unet=unet, torch_dtype=dtype, variant="fp16").to(device)
+    pipe = AnimateDiffPipeline.from_pretrained(base, motion_adapter=adapter, unet=unet, torch_dtype=dtype, variant="fp16", device_map=device)
     pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", beta_schedule="linear")
     
     mp.set_start_method("spawn", force=True)