From 11afb67860d406df860d32a22fc1b31fd96960c7 Mon Sep 17 00:00:00 2001
From: Rockford Mankini <rmankini@ucsd.edu>
Date: Wed, 14 Feb 2024 09:16:15 -0800
Subject: [PATCH] add transformers to scipy ml

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

diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile
index f12bb9bc..3f650f64 100644
--- a/images/scipy-ml-notebook/Dockerfile
+++ b/images/scipy-ml-notebook/Dockerfile
@@ -91,6 +91,8 @@ 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 && pip cache purge
+
 USER $NB_UID:$NB_GID
 ENV PATH=${PATH}:/usr/local/nvidia/bin:/opt/conda/bin