diff --git a/.ci/torchbench/install.sh b/.ci/torchbench/install.sh
index 4828b6711..16dcfc6e8 100644
--- a/.ci/torchbench/install.sh
+++ b/.ci/torchbench/install.sh
@@ -26,4 +26,5 @@ parent_dir=$(dirname "$(readlink -f "$0")")/../..
 cd ${parent_dir}
 
 python -c "import torch; print(torch.__version__); print(torch.version.git_version)"
-python install.py
+
+python install.py $@