diff --git a/src/axolotl/utils/samplers/multipack.py b/src/axolotl/utils/samplers/multipack.py index 451893971..1acaa51b9 100644 --- a/src/axolotl/utils/samplers/multipack.py +++ b/src/axolotl/utils/samplers/multipack.py @@ -181,7 +181,7 @@ class MultipackBatchSampler(BatchSampler): ) # shave off 1% + 1 for dealing with variance in packing from random sampler to sampler - return min( + return max( 1, ( world_size