diff --git a/docker-cuda/Dockerfile_base_finetune b/docker-cuda/Dockerfile_base_finetune index c93079d5..843fdb06 100644 --- a/docker-cuda/Dockerfile_base_finetune +++ b/docker-cuda/Dockerfile_base_finetune @@ -1,6 +1,6 @@ # This dockerfile is meant to serve as a rocm base image. It registers the debian rocm package repository, and # installs the rocm-dev package. -FROM ebolam/koboldai_base +FROM ebolam/koboldai_base:bare WORKDIR /content/ 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 39b569c1..c3fb8833 100644 --- a/docker-cuda/Dockerfile_base_huggingface +++ b/docker-cuda/Dockerfile_base_huggingface @@ -2,6 +2,6 @@ # installs the rocm-dev package. -FROM ebolam/koboldai_base +FROM ebolam/koboldai_base:bare WORKDIR /content/ RUN micromamba update -y -n base -f /content/environments/huggingface.yml