fix error

Mrz1526 · Jan 9, 2024 · eb5e048 · eb5e048
1 parent a74e634
commit eb5e048
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/dataset.py b/utils/dataset.py
@@ -132,7 +132,7 @@ def __init__(
             vid_data_key: str = "video_path",
             preprocessed: bool = False,
             use_bucketing: bool = False,
-            cache_latents: = False,
+            cache_latents: bool = False,
             motion_threshold = 50,
             **kwargs
     ):
@@ -226,7 +226,7 @@ def train_data_batch(self, index):
             "pixel_values": normalize_input(video),
             "prompt_ids": prompt_ids,
             "text_prompt": prompt,
-            'dataset': self.__getname__()
+            'dataset': self.__getname__(),
             'cache_path': cache_path,
         }
         mask = get_moved_area_mask(video.permute([0,2,3,1]).numpy())