[Misc] Add packages for benchmark as extra dependency (#19089)

Signed-off-by: Isotr0py <2037008807@qq.com>
2025-06-04 19:18:48 +08:00
parent 2669a0d7b5
commit 8711bc5e68
3 changed files with 20 additions and 22 deletions
--- a/setup.py
+++ b/setup.py
@ -688,6 +688,7 @@ setup(
    ext_modules=ext_modules,
    install_requires=get_requirements(),
    extras_require={
+        "bench": ["pandas", "datasets"],
        "tensorizer": ["tensorizer>=2.9.0"],
        "fastsafetensors": ["fastsafetensors >= 0.1.10"],
        "runai": ["runai-model-streamer", "runai-model-streamer-s3", "boto3"],