diff --git a/docker-cuda/Dockerfile_base_finetune b/docker-cuda/Dockerfile_base_finetune index 0b67a7db..c93079d5 100644 --- 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 diff --git a/docker-cuda/Dockerfile_base_huggingface b/docker-cuda/Dockerfile_base_huggingface index 47406cc1..39b569c1 100644 --- 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