From 3179b435e4eb073eefe1913d0e166f976bc50340 Mon Sep 17 00:00:00 2001 From: Marc Sun Date: Mon, 17 Jun 2024 15:50:01 +0200 Subject: [PATCH] Fix arg in accelerate config --- src/transformers/trainer.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index 34cf5aa49046..17bdefb0516c 100755 --- a/src/transformers/trainer.py +++ b/src/transformers/trainer.py @@ -4544,6 +4544,9 @@ def create_accelerator_and_postprocess(self): even_batches=accelerator_config.pop("even_batches"), use_seedable_sampler=accelerator_config.pop("use_seedable_sampler"), ) + if not is_accelerate_available("0.26.0"): + accelerator_config.pop("use_seedable_sampler") + non_blocking = accelerator_config.pop("non_blocking") if not is_accelerate_available("0.30.0"): if non_blocking: