Update Dockerfile

logikon-ai · Aug 1, 2024 · 50e6d7e · 50e6d7e
1 parent b62ea5c
commit 50e6d7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -16,7 +16,7 @@ RUN git clone --branch ${LM_EVAL_VERSION} https://github.com/EleutherAI/lm-evalu
 # Install python packages
 
 RUN pip install --upgrade pip
-RUN pip uninstall transformer-engine -y
+#RUN pip uninstall transformer-engine -y
 
 RUN cd lm-evaluation-harness && pip install -e .
 RUN cd cot-eval && pip install -e .