diff --git a/benchmarks/accuracy/fastchat.sh b/benchmarks/accuracy/fastchat.sh index 3d1c953..334b34e 100644 --- a/benchmarks/accuracy/fastchat.sh +++ b/benchmarks/accuracy/fastchat.sh @@ -25,7 +25,8 @@ function install_fastchat { git clone https://github.com/AlibabaPAI/FastChat_TorchAcc.git fi - if python -c "import fschat" &>/dev/null; then + output=$(python -m pip list | grep fschat) + if [[ -n $output ]]; then echo "All requirements are installed." else echo "Install requirements ..." diff --git a/benchmarks/accuracy/run.sh b/benchmarks/accuracy/run.sh index 1e32d2b..94bbd12 100755 --- a/benchmarks/accuracy/run.sh +++ b/benchmarks/accuracy/run.sh @@ -104,5 +104,5 @@ function collect_and_show_results { do_train do_evaluation -upload_to_oss collect_and_show_results +upload_to_oss