adding hf_transfer

ROCm · Feb 20, 2025 · d35adbb · d35adbb
1 parent 39456f3
commit d35adbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.rocm b/Dockerfile.rocm
@@ -87,7 +87,7 @@ RUN case "$(which python3)" in \
             rm -rf /opt/conda/envs/py_3.9/lib/python3.9/site-packages/numpy-1.20.3.dist-info/;; \
         *) ;; esac
 
-RUN python3 -m pip install --upgrade huggingface-hub[cli]
+RUN python3 -m pip install --upgrade huggingface-hub[cli,hf_transfer]
 ARG BUILD_RPD
 RUN if [ ${BUILD_RPD} -eq "1" ]; then \
     git clone -b nvtx_enabled https://github.com/ROCm/rocmProfileData.git \