rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-01-09 06:48:23 +01: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
Show Stats 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
WORKDIR /content/
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
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
# installs the rocm-dev package.
FROM mambaorg/micromamba
FROM ebolam/koboldai_base
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
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

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