.

options/Video_model/Model.py

@@ -47,6 +47,11 @@ def Video(
     base_count = len(glob(os.path.join(output_folder, "*.mp4")))
     video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
 
+    if isinstance(image, torch.Tensor):
+        # Convert BFloat16 to Float32 if necessary
+        if image.dtype == torch.bfloat16:
+            image = image.to(dtype=torch.float32)
+
     with torch.autocast(device):
         frames = pipeline(
             image, height = height, width = width,