reformat black

huggingface · Oct 11, 2023 · 5c9ace2 · 5c9ace2
1 parent d61f046
commit 5c9ace2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/optimum/onnxruntime/trainer.py b/optimum/onnxruntime/trainer.py
@@ -788,7 +788,8 @@ def _inner_training_loop(
                             model.clip_grad_norm_(args.max_grad_norm)
                         else:
                             self.accelerator.clip_grad_norm_(
-                                model.parameters(), args.max_grad_norm,
+                                model.parameters(),
+                                args.max_grad_norm,
                             )
 
                     # Optimizer step