Update dataset.py

ExponentialML · web-flow · commit f66580f2854d · 2023-03-30T12:10:23.000-07:00
diff --git a/utils/dataset.py b/utils/dataset.py
@@ -211,6 +211,7 @@ def __init__(
         fps: int = 8,
         path: str = "./data",
         fallback_prompt: str = "",
+        **kwargs
     ):
         self.tokenizer = tokenizer
 
