better video quality

baseline_utils.py

@@ -127,7 +127,7 @@ def generate_video(activity_list, writer_summary, fps=24):  # Lower fps
     os.makedirs("videos", exist_ok=True)
     video_paths = []
     for i, prompt in enumerate(prompts):
-        video_frames = pipe(prompt, num_inference_steps=10, height=320, width=576, num_frames=fps).frames
+        video_frames = pipe(prompt, num_inference_steps=40, height=320, width=576, num_frames=fps).frames
         video_path = export_to_video(np.squeeze(video_frames, axis=0), output_video_path=f'videos/video{i}.mp4')
         video_paths.append(video_path)