diff --git a/tools/ci_build/github/pai/pai_huggingface_bert_large_test.sh b/tools/ci_build/github/pai/pai_huggingface_bert_large_test.sh
index ee1f0482120c7..3b8e828fe94dd 100644
--- a/tools/ci_build/github/pai/pai_huggingface_bert_large_test.sh
+++ b/tools/ci_build/github/pai/pai_huggingface_bert_large_test.sh
@@ -3,7 +3,7 @@
 set -ex
 
 rocm_version=$1
-mi200_gpus=$(rocm-smi --showproductname | grep -c "MI250")
+mi200_gpus=$(rocm-smi --showproductname | grep -c "MI250" | xargs)
 
 echo "mi200_gpus: $mi200_gpus"