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 base image again to only have transformers change between the images

Browse Source
This commit is contained in:
ebolam
2022-01-26 15:35:36 -05:00
parent d2b15e2a6e
commit a27c441cdf
3 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker-cuda/Dockerfile_base
View File

@ -5,6 +5,6 @@
FROM mambaorg/micromamba FROM mambaorg/micromamba
WORKDIR /content/ WORKDIR /content/
COPY environments /content/environments COPY environments /content/environments
RUN micromamba install -y -n base -f /content/environments/huggingface.yml RUN micromamba install -y -n base -f /content/environments/base.yml
USER root USER root
RUN apt update && apt install xorg -y RUN apt update && apt install xorg -y

9
docker-cuda/Dockerfile_base_finetune
View File

@ -1,9 +1,8 @@
# This dockerfile is meant to serve as a rocm base image. It registers the debian rocm package repository, and # This dockerfile is meant to serve as a rocm base image. It registers the debian rocm package repository, and
# installs the rocm-dev package. # installs the rocm-dev package.
FROM mambaorg/micromamba FROM ebolam/koboldai_base
WORKDIR /content/ WORKDIR /content/
COPY environments /content/environments RUN apt-get update
RUN micromamba install -y -n base -f /content/environments/finetuneanon.yml RUN apt-get install git -y
USER root RUN /opt/conda/bin/pip install tensorflow-gpu git+https://github.com/finetuneanon/transformers.git
RUN apt update && apt install xorg -y

7
docker-cuda/Dockerfile_base_huggingface Normal file
View File

@ -0,0 +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 ebolam/koboldai_base
WORKDIR /content/
RUN /opt/conda/bin/pip install transformers