Docker file fix

2022-01-26 21:14:10 -05:00 · 2022-01-26 21:14:10 -05:00 · ab5d3b4255
parent 2278b7c103
commit ab5d3b4255
2 changed files with 2 additions and 4 deletions
--- a/docker-cuda/Dockerfile_base_finetune
+++ b/docker-cuda/Dockerfile_base_finetune
@ -3,6 +3,4 @@

 FROM ebolam/koboldai_base
 WORKDIR /content/
-RUN apt-get update
-RUN apt-get install git -y
-RUN /opt/conda/bin/pip install tensorflow-gpu git+https://github.com/finetuneanon/transformers.git
+RUN micromamba update -y -n base -f /content/environments/finetuneanon.yml
--- a/docker-cuda/Dockerfile_base_huggingface
+++ b/docker-cuda/Dockerfile_base_huggingface
@ -4,4 +4,4 @@

 FROM ebolam/koboldai_base
 WORKDIR /content/
-RUN /opt/conda/bin/pip install transformers
+RUN micromamba update -y -n base -f /content/environments/huggingface.yml