normalize video_to_tensor output, thanks to @iejMac

lucidrains · Nov 26, 2023 · dfd79cf · dfd79cf
1 parent b2f105b
commit dfd79cf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/magvit2_pytorch/data.py b/magvit2_pytorch/data.py
@@ -183,6 +183,8 @@ def video_to_tensor(
 
     frames_torch = torch.tensor(frames).float()
 
+    frames_torch /= 255.
+
     return frames_torch[:, :num_frames, :, :]
 
 @beartype

diff --git a/magvit2_pytorch/version.py b/magvit2_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.1.40'
+__version__ = '0.1.41'