rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Updated dockerfiles to create images for docker hub that are per-compiled

Browse Source
This commit is contained in:
ebolam
2022-01-26 11:35:58 -05:00
parent b0f1bdf2fd
commit d2b15e2a6e
4 changed files with 23 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docker-cuda/Dockerfile
View File

@@ -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
CMD python aiserver.py

10
docker-cuda/Dockerfile_base Normal file
View File

@@ -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

9
docker-cuda/Dockerfile_base_finetune Normal file
View File

@@ -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

8
docker-cuda/Dockerfile_finetune
View File

@@ -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
CMD python aiserver.py