enable hf trasfer and add unzip to image

2023-10-29 04:53:14 -04:00
parent 827ec3d274
commit db86e32cf4
2 changed files with 2 additions and 1 deletions
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -8,6 +8,7 @@ ENV BNB_CUDA_VERSION=$CUDA
 ARG PYTORCH_VERSION="2.0.1"

 ENV PYTORCH_VERSION=$PYTORCH_VERSION
+ENV HF_HUB_ENABLE_HF_TRANSFER=1

 RUN apt-get update && \
    apt-get install -y vim curl