diff --git a/Dockerfile b/Dockerfile index 343364da2ebf5..922183e21444d 100644 --- a/Dockerfile +++ b/Dockerfile @@ -48,6 +48,8 @@ COPY requirements-common.txt requirements-common.txt COPY requirements-cuda.txt requirements-cuda.txt RUN --mount=type=cache,target=/root/.cache/pip \ python3 -m pip install -r requirements-cuda.txt +RUN --mount=type=cache,target=/root/.cache/pip \ + python3 -m pip install hf_transfer # cuda arch list used by torch