[Benchmark] Reuse optimum-benchmark (#30615)

* benchmark

* update

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2024-05-21 15:15:19 +02:00
committed by GitHub
parent 3b09d3f05f
commit 64e0573a81
7 changed files with 393 additions and 1 deletions

View File

@@ -136,6 +136,7 @@ _deps = [
"onnxruntime-tools>=1.4.2",
"onnxruntime>=1.4.0",
"opencv-python",
"optimum-benchmark>=0.2.0",
"optuna",
"optax>=0.0.8,<=0.1.4",
"packaging>=20.0",
@@ -410,6 +411,8 @@ extras["agents"] = deps_list(
"diffusers", "accelerate", "datasets", "torch", "sentencepiece", "opencv-python", "Pillow"
)
extras["benchmark"] = deps_list("optimum-benchmark")
# when modifying the following list, make sure to update src/transformers/dependency_versions_check.py
install_requires = [
deps["filelock"], # filesystem locks, e.g., to prevent parallel downloads