Removed TensorFlow dependency

neel04 · Nov 23, 2024 · 43574c8 · 43574c8
1 parent 8067e74
commit 43574c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -20,7 +20,7 @@ RUN pip3 install -q transformers datasets scalax tokenizers icecream wandb einop
 RUN pip3 install -U optuna-integration plotly lm-eval pdbpp
 RUN pip3 install git+https://github.com/deepmind/jmp
 RUN pip3 install git+https://github.com/Findus23/jax-array-info.git
-RUN pip3 install -q tensorflow tensorboard-plugin-profile "cloud-tpu-profiler>=2.3.0"
+RUN pip3 install -q tensorboard-plugin-profile "cloud-tpu-profiler>=2.3.0"
 
 WORKDIR /ReAct_Jax