ugh

stanford-crfm · Sep 13, 2024 · e33a905 · e33a905
1 parent b6f334e
commit e33a905
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/levanter/data/text.py b/src/levanter/data/text.py
@@ -600,7 +600,7 @@ def train_set(
 
         if self.shuffle is True:
             ds = ds.shuffle(key)
-        elif isinstance(self.shuffle, int):
+        elif isinstance(self.shuffle, int) and self.shuffle > 0:
             ds = ds.era_shuffle(self.shuffle, key=key)
 
         return ds  # type: ignore