From a38ee328bda21e6b2ce9c3152e97b0c7772eaa5a Mon Sep 17 00:00:00 2001
From: RockfordMankiniUCSD
 <92760584+RockfordMankiniUCSD@users.noreply.github.com>
Date: Mon, 1 Apr 2024 16:24:08 -0700
Subject: [PATCH] Update Dockerfile (#87)

---
 images/scipy-ml-notebook/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile
index 5c709032..afd0d265 100644
--- a/images/scipy-ml-notebook/Dockerfile
+++ b/images/scipy-ml-notebook/Dockerfile
@@ -94,7 +94,7 @@ RUN pip install torch==2.1.2 torchvision torchaudio --index-url https://download
   rm /opt/conda/lib/python3.9/site-packages/torch/lib/libcudnn_ops_train.so.8 && \
   rm /opt/conda/lib/python3.9/site-packages/torch/lib/libcublas.so.11
 
-RUN pip install transformers datasets accelerate huggingface-cli && pip cache purge
+RUN pip install transformers datasets accelerate huggingface-cli timm && pip cache purge
 
 USER $NB_UID:$NB_GID
 ENV PATH=${PATH}:/usr/local/nvidia/bin:/opt/conda/bin