video_tensor_to_gif typo fix

kyegomez · Dec 29, 2023 · bdf2090 · bdf2090
1 parent bd81992
commit bdf2090
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/zeta/utils/main.py b/zeta/utils/main.py
@@ -458,7 +458,7 @@ def seek_all_images(img, channels=3):
 
 # tensor of shape (channels, frames, height, width) -> GIF
 def video_tensor_to_gift(tensor, path, duration=120, loop=0, optimize=True):
- images = map(T.ToPilImage(), tensor.unbind(dim=1))
+ images = map(T.ToPILImage(), tensor.unbind(dim=1))
  first_img, *rest_imgs = images
  first_img.save(
  path,