Update tuneavideo/util.py

tuneavideo/util.py

@@ -14,7 +14,7 @@ def save_videos_grid(
     path: str = 'output.gif', 
     rescale=False, 
     n_rows=4, 
-    fps=3
+    duration=3
 ):
     videos = rearrange(videos, "b c t h w -> t b c h w")
     outputs = []
@@ -30,5 +30,5 @@ def save_videos_grid(
         os.makedirs(save_path)
     
     
-    imageio.mimsave(os.path.join(save_path, path), outputs, fps=fps)
+    imageio.mimsave(os.path.join(save_path, path), outputs, duration=duration)
     return os.path.join(save_path, path)