diff --git a/docker-cuda/Dockerfile b/docker-cuda/Dockerfile index e863ea58..36dac8e1 100644 --- a/docker-cuda/Dockerfile +++ b/docker-cuda/Dockerfile @@ -1,11 +1,7 @@ # 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 mambaorg/micromamba +FROM ebolam/koboldai_base WORKDIR /content/ COPY . /content/ -RUN ls -RUN micromamba install -y -n base -f /content/environments/huggingface.yml -USER root -RUN apt update && apt install xorg -y -CMD python aiserver.py \ No newline at end of file +CMD python aiserver.py diff --git a/docker-cuda/Dockerfile_base b/docker-cuda/Dockerfile_base new file mode 100644 index 00000000..c026f9df --- /dev/null +++ b/docker-cuda/Dockerfile_base @@ -0,0 +1,10 @@ +# 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 mambaorg/micromamba +WORKDIR /content/ +COPY environments /content/environments +RUN micromamba install -y -n base -f /content/environments/huggingface.yml +USER root +RUN apt update && apt install xorg -y diff --git a/docker-cuda/Dockerfile_base_finetune b/docker-cuda/Dockerfile_base_finetune new file mode 100644 index 00000000..4fd7d0ff --- /dev/null +++ b/docker-cuda/Dockerfile_base_finetune @@ -0,0 +1,9 @@ +# 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 mambaorg/micromamba +WORKDIR /content/ +COPY environments /content/environments +RUN micromamba install -y -n base -f /content/environments/finetuneanon.yml +USER root +RUN apt update && apt install xorg -y diff --git a/docker-cuda/Dockerfile_finetune b/docker-cuda/Dockerfile_finetune index 39e13d98..cd8f4b4b 100644 --- a/docker-cuda/Dockerfile_finetune +++ b/docker-cuda/Dockerfile_finetune @@ -1,11 +1,7 @@ # 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 mambaorg/micromamba +FROM ebolam/koboldai_base:finetune WORKDIR /content/ COPY . /content/ -RUN ls -RUN micromamba install -y -n base -f /content/environments/finetuneanon.yml -USER root -RUN apt update && apt install xorg -y -CMD python aiserver.py \ No newline at end of file +CMD python aiserver.py