From ab5d3b42556d6cb0b47212ea73349286d6c67b3b Mon Sep 17 00:00:00 2001 From: ebolam Date: Wed, 26 Jan 2022 21:14:10 -0500 Subject: [PATCH] Docker file fix --- docker-cuda/Dockerfile_base_finetune | 4 +--- docker-cuda/Dockerfile_base_huggingface | 2 +- 2 files changed, 2 insertions(+), 4 deletions(-) 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