.

options/Video_model/utils.py

@@ -29,9 +29,14 @@ def load_lora_weights(unet, pretrained_model_name_or_path_or_dict: Union[str, Di
         adapter_name=adapter_name,
     )
 
+
 def save_video(frames, save_path, fps, quality=9):
+    # Ensure the file is saved with a video format
+    if not save_path.endswith(('.mp4', '.avi', '.gif')):
+        save_path += '.mp4'  # Default to mp4 if no valid extension is provided
+
     writer = imageio.get_writer(save_path, fps=fps, quality=quality)
     for frame in frames:
         frame = np.array(frame)
         writer.append_data(frame)
-    writer.close()
+    writer.close()