mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-22 16:17:57 +01:00
26 lines
1.1 KiB
Docker
26 lines
1.1 KiB
Docker
|
# BUILDER
|
||
|
|
FROM ubuntu:22.04
|
||
|
|
WORKDIR /builder
|
||
|
|
ARG TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST:-3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX}"
|
||
|
|
ARG BUILD_EXTENSIONS="${BUILD_EXTENSIONS:-}"
|
||
|
|
ARG APP_UID="${APP_UID:-6972}"
|
||
|
|
ARG APP_GID="${APP_GID:-6972}"
|
||
|
|
ARG GPU_CHOICE=A
|
||
|
|
ARG USE_CUDA118=FALSE
|
||
|
|
ARG LAUNCH_AFTER_INSTALL=FALSE
|
||
|
|
ARG INSTALL_EXTENSIONS=TRUE
|
||
|
|
|
||
|
|
RUN --mount=type=cache,target=/var/cache/apt,sharing=locked,rw \
|
||
|
|
apt update && \
|
||
|
|
apt install --no-install-recommends -y git vim build-essential python3-dev pip bash curl && \
|
||
|
|
rm -rf /var/lib/apt/lists/*
|
||
|
|
WORKDIR /home/app/
|
||
|
|
RUN git clone https://github.com/oobabooga/text-generation-webui.git
|
||
|
|
WORKDIR /home/app/text-generation-webui
|
||
|
|
RUN GPU_CHOICE=N USE_CUDA118=FALSE LAUNCH_AFTER_INSTALL=FALSE INSTALL_EXTENSIONS=TRUE ./start_linux.sh --verbose
|
||
|
|
COPY CMD_FLAGS.txt /home/app/text-generation-webui/
|
||
|
|
EXPOSE ${CONTAINER_PORT:-7860} ${CONTAINER_API_PORT:-5000} ${CONTAINER_API_STREAM_PORT:-5005}
|
||
|
|
# set umask to ensure group read / write at runtime
|
||
|
|
WORKDIR /home/app/text-generation-webui
|
||
|
|
CMD umask 0002 && export HOME=/home/app/text-generation-webui && ./start_linux.sh
|