From 4b30dbbd46bd213c15489ccf10411d542fc3f252 Mon Sep 17 00:00:00 2001 From: Dominic Feliton <127802411+dafeliton@users.noreply.github.com> Date: Thu, 22 Aug 2024 14:20:24 -0700 Subject: [PATCH] Try using huggingface-hub instead --- images/scipy-ml-notebook/Dockerfile | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile index e31b1769..265f0702 100644 --- a/images/scipy-ml-notebook/Dockerfile +++ b/images/scipy-ml-notebook/Dockerfile @@ -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