Update app.py

app.py

@@ -19,7 +19,8 @@ image_pipe = StableDiffusionPipeline.from_pretrained(
 ).to(device)
 
 video_pipe = StableVideoDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-video-diffusion-img2vid", torch_dtype=torch.float32
+    "stabilityai/stable-video-diffusion-img2vid", 
+    torch_dtype=torch.float16 if device == "cuda" else torch.float32
 ).to(device)
 
 music_model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")