fix print in benchmark (#5242)

b2968373 · Patrick von Platen · GitHub · 9fe09cec · b2968373
Unverified Commit b2968373 authored Jun 24, 2020 by Patrick von Platen Committed by GitHub Jun 24, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/benchmark/benchmark_utils.py src/transformers/benchmark/benchmark_utils.py +1 -1

No files found.
--- a/src/transformers/benchmark/benchmark_utils.py
+++ b/src/transformers/benchmark/benchmark_utils.py
@@ -807,7 +807,7 @@ class Benchmark(ABC):
                    else:
                        result = str(result)
                    self.print_fn(
-                        model_name.center(30) + str(batch_size).center(15),
+                        model_name[:30].center(30) + str(batch_size).center(15),
                        str(sequence_length).center(15),
                        result.center(15),
                    )