fps=2

test_vid2vid_zero.py

@@ -83,7 +83,7 @@ def main(
     use_sc_attn: bool = True,
     use_st_attn: bool = True,
     st_attn_idx: int = 0,
-    fps: int = 1,
+    fps: int = 2,
 ):
     *_, config = inspect.getargvalues(inspect.currentframe())
 

vid2vid_zero/util.py

@@ -12,7 +12,7 @@ from tqdm import tqdm
 from einops import rearrange
 
 
-def save_videos_as_images(videos: torch.Tensor, path: str, rescale=False, n_rows=4, fps=1):
+def save_videos_as_images(videos: torch.Tensor, path: str, rescale=False, n_rows=4, fps=2):
     dir_name = os.path.dirname(path)
     videos = rearrange(videos, "b c t h w -> t b h w c")
 
@@ -30,7 +30,7 @@ def save_videos_as_images(videos: torch.Tensor, path: str, rescale=False, n_rows
             image.save(save_path)
 
 
-def save_videos_grid(videos: torch.Tensor, path: str, rescale=False, n_rows=4, fps=1):
+def save_videos_grid(videos: torch.Tensor, path: str, rescale=False, n_rows=4, fps=2):
     videos = rearrange(videos, "b c t h w -> t b c h w")
     outputs = []
     for x in videos: