Add hf_transfer to testing image

neuralmagic · Nov 6, 2024 · 151a609 · 151a609
1 parent 098f94d
commit 151a609
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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