Typo

ExponentialML · web-flow · commit da46f929cb32 · 2023-04-09T14:59:37.000-07:00
diff --git a/utils/dataset.py b/utils/dataset.py
@@ -557,7 +557,7 @@ def __getitem__(self, index):
 
         prompt_ids = self.get_prompt_ids(prompt)
 
-        return {"pixel_values": (video[0 / 127.5 - 1.0), "prompt_ids": prompt_ids[0], "text_prompt": prompt, 'dataset': self.__getname__()}
+        return {"pixel_values": (video[0] / 127.5 - 1.0), "prompt_ids": prompt_ids[0], "text_prompt": prompt, 'dataset': self.__getname__()}
 
 class CachedDataset(Dataset):
     def __init__(self,cache_dir: str = ''):
