Install huggingface_hub package for gpu tests.

SamanehSaadat · Apr 16, 2024 · a8845ea · a8845ea
1 parent 40a0e11
commit a8845ea
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.kokoro/github/ubuntu/gpu/build.sh b/.kokoro/github/ubuntu/gpu/build.sh
@@ -57,6 +57,7 @@ then
 fi
 
 pip install --no-deps -e "." --progress-bar off
+pip install huggingface_hub
 
 # Run Extra Large Tests for Continuous builds
 if [ "${RUN_XLARGE:-0}" == "1" ]