correct paths

baseline_utils.py

@@ -147,20 +147,20 @@ def generate_video(scene_list, fps=24):  # Lower fps
     # Truncate the prompt to fit the CLIP token limit
     os.makedirs("videos", exist_ok=True)
     video_paths = []
-    #for i, prompt in enumerate(scene_list):
-    video = pipe(
-        prompt=scene_list,
-        num_videos_per_prompt=1,
-        num_inference_steps=1,
-        num_frames=fps,
-        guidance_scale=6,
-        generator=torch.Generator(device=device).manual_seed(42),
-    ).frames[0]
-
-    video_path = export_to_video(video)
-        #video_paths.append(video_path)
-
-    return video_path
+    for i, prompt in enumerate(scene_list):
+        video = pipe(
+            prompt=prompt,
+            num_videos_per_prompt=1,
+            num_inference_steps=40,
+            num_frames=fps,
+            guidance_scale=6,
+            generator=torch.Generator(device=device).manual_seed(42),
+        ).frames[0]
+
+        video_path = export_to_video(video, output_video_path=f'videos/video{i}.mp4')
+        video_paths.append(video_path)
+
+    return video_paths