fix dtype bug on CPU (#1)

- fix dtype bug on CPU (bff0f0a3074e9a74a99037e7c1c815296606b738)


Co-authored-by: Joyce Lin <jycln@users.noreply.huggingface.co>

app.py

@@ -15,11 +15,13 @@ if st.button("🎬 Generate Video"):
         device = "cuda" if torch.cuda.is_available() else "cpu"
         dtype = torch.float16 if device == "cuda" else torch.float32
 
+        use_fp16 = torch.cuda.is_available()
+
         pipe = DiffusionPipeline.from_pretrained(
             "damo-vilab/text-to-video-ms-1.7b",
-            torch_dtype=dtype,
-            variant="fp16" if device == "cuda" else None,
-        ).to(device)
+            torch_dtype=torch.float16 if use_fp16 else torch.float32,
+            variant="fp16" if use_fp16 else None,
+        ).to("cuda" if use_fp16 else "cpu")
 
         video_frames = pipe(prompt).frames  # Returns list of PIL Images (frames)