Wipe cache if wrong len pt2

Author: daviswer

fms_fsdp/utils/dataset_utils.py

@@ -706,7 +706,7 @@ def _pad_buffer(self):
         if len(self.buffer) < self.window_size:
             self.buffer += [
                 [],
-            ] * (len(self.buffer) - self.buffer_size)
+            ] * (self.window_size - len(self.buffer))
 
     def state_dict(self):
         # Write generator state manually