| # 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 bash-completion && \ | |
| 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 --model /Users/computer/git/text-generation-webui/models/LLamandementFineTunéWithoutNotation16Q.gguf --loader llama.cpp --verbose | |