fix print in benchmark (huggingface#5242)

zsc · Jun 24, 2020 · b296837 · b296837
1 parent 9fe09ce
commit b296837
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/benchmark/benchmark_utils.py b/src/transformers/benchmark/benchmark_utils.py
@@ -807,7 +807,7 @@ def print_results(self, result_dict, type_label):
                     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),
                     )