update

src/transformers/benchmark/benchmark_utils_generic.py

@@ -73,7 +73,10 @@ class BenchMark:
             return report.__name__
         return report
 
-    def _report(self, result, only_result=False, output_path=None):
+    def _report(self, result, output_path=None, only_result=False):
+        self._run_buffer["config"]["report_kwargs"]["output_path"] = output_path
+        self._run_buffer["config"]["report_kwargs"]["only_result"] = only_result
+
         self._run_buffer["result"] = result
         self._buffer["runs"].append(self._run_buffer)
 

src/transformers/benchmark/from_pretrained_benchmark.py

@@ -21,6 +21,7 @@ from benchmark_utils_generic import BenchMark, SpeedBenchMark
 
 import transformers
 
+
 class FromPretrainedBenchMark(BenchMark):
     def _target(self, model_class, repo_id):
         self._run_buffer["config"]["target_kwargs"]["model_class"] = model_class