Updated dockerfile with lm-eval

neel04 · Oct 23, 2024 · 701cac5 · 701cac5
1 parent 96fbec7
commit 701cac5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -18,7 +18,7 @@ RUN pip3 install numpy pandas scipy
 RUN pip3 install -U numpy==1.26.4
 RUN pip3 install -U -q jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
 RUN pip3 install -q transformers datasets scalax tokenizers icecream wandb einops torch tqdm jaxtyping optax optuna equinox rich
-RUN pip3 install -U tensorboard-plugin-profile optuna-integration plotly
+RUN pip3 install -U tensorboard-plugin-profile optuna-integration plotly lm-eval
 RUN pip3 install git+https://github.com/deepmind/jmp
 RUN pip3 install git+https://github.com/Findus23/jax-array-info.git