oops

stanford-crfm · Feb 9, 2025 · 92c9f1d · 92c9f1d
1 parent 0ce91d6
commit 92c9f1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/levanter/data/loader.py b/src/levanter/data/loader.py
@@ -215,7 +215,7 @@ def __init__(self, data_loader: DataLoader, start_from_batch: Optional[int] = No
 
         buffered_batches = self.dl.max_buffered_batches
         self._batches: Iterator[Ex]
-        if buffered_batches == 0 or True:
+        if buffered_batches == 0:
             self._batches = AsyncIteratorWrapper(self._produce_batches())
         else:
             self._batches = _JaxCpuBackgroundIterator(self._produce_batches, max_capacity=buffered_batches)