Skip to content

Commit

Permalink
Clean
Browse files Browse the repository at this point in the history
  • Loading branch information
muellerzr committed Jul 1, 2024
1 parent 4a81c8c commit 6cb8128
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -4645,7 +4645,7 @@ def create_accelerator_and_postprocess(self):

# `auto_find_batch_size` isn't yet supported with DeepSpeed
if self.is_deepspeed_enabled and self.args.auto_find_batch_size:
raise NotImplementedError(f"`DeepSpeed` doesn't support `auto_find_batch_size`.")
raise NotImplementedError("`DeepSpeed` doesn't support `auto_find_batch_size`.")

def propagate_args_to_deepspeed(self, auto_find_batch_size=False):
"""
Expand Down

0 comments on commit 6cb8128

Please sign in to comment.