Skip to content

Commit

Permalink
Try using huggingface-hub instead
Browse files Browse the repository at this point in the history
  • Loading branch information
dafeliton authored Aug 22, 2024
1 parent 4fd732e commit 4b30dbb
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion images/scipy-ml-notebook/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,11 @@ RUN pip install nvidia-cudnn-cu12==$CUDNN_VERSION torch==$TORCH_VERSION torchvis
mamba clean -a -y && \
pip cache purge

RUN pip install transformers datasets accelerate huggingface-cli timm && pip cache purge
RUN pip install transformers datasets accelerate huggingface-hub timm && \
fix-permissions $CONDA_DIR && \
fix-permissions /home/$NB_USER && \
mamba clean -a -y && \
pip cache purge

USER $NB_UID:$NB_GID
ENV PATH=${PATH}:/usr/local/nvidia/bin:/opt/conda/bin
Expand Down

0 comments on commit 4b30dbb

Please sign in to comment.