mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2025-06-07 06:06:20 -04:00
Fix Dockerfile in AMD and Intel (#6995)
This commit is contained in:
parent
83bfd5c64b
commit
9cd6ea6c0b
5 changed files with 6 additions and 24 deletions
|
@ -146,14 +146,14 @@ The `requirements*.txt` above contain various wheels precompiled through GitHub
|
||||||
For NVIDIA GPU:
|
For NVIDIA GPU:
|
||||||
ln -s docker/{nvidia/Dockerfile,nvidia/docker-compose.yml,.dockerignore} .
|
ln -s docker/{nvidia/Dockerfile,nvidia/docker-compose.yml,.dockerignore} .
|
||||||
For AMD GPU:
|
For AMD GPU:
|
||||||
ln -s docker/{amd/Dockerfile,intel/docker-compose.yml,.dockerignore} .
|
ln -s docker/{amd/Dockerfile,amd/docker-compose.yml,.dockerignore} .
|
||||||
For Intel GPU:
|
For Intel GPU:
|
||||||
ln -s docker/{intel/Dockerfile,amd/docker-compose.yml,.dockerignore} .
|
ln -s docker/{intel/Dockerfile,amd/docker-compose.yml,.dockerignore} .
|
||||||
For CPU only
|
For CPU only
|
||||||
ln -s docker/{cpu/Dockerfile,cpu/docker-compose.yml,.dockerignore} .
|
ln -s docker/{cpu/Dockerfile,cpu/docker-compose.yml,.dockerignore} .
|
||||||
cp docker/.env.example .env
|
cp docker/.env.example .env
|
||||||
#Create logs/cache dir :
|
#Create logs/cache dir :
|
||||||
mkdir -p logs cache
|
mkdir -p user_data/logs user_data/cache
|
||||||
# Edit .env and set:
|
# Edit .env and set:
|
||||||
# TORCH_CUDA_ARCH_LIST based on your GPU model
|
# TORCH_CUDA_ARCH_LIST based on your GPU model
|
||||||
# APP_RUNTIME_GID your host user's group id (run `id -g` in a terminal)
|
# APP_RUNTIME_GID your host user's group id (run `id -g` in a terminal)
|
||||||
|
|
|
@ -14,7 +14,7 @@ WORKDIR /home/app/
|
||||||
RUN git clone https://github.com/oobabooga/text-generation-webui.git
|
RUN git clone https://github.com/oobabooga/text-generation-webui.git
|
||||||
WORKDIR /home/app/text-generation-webui
|
WORKDIR /home/app/text-generation-webui
|
||||||
RUN GPU_CHOICE=B LAUNCH_AFTER_INSTALL=FALSE INSTALL_EXTENSIONS=TRUE ./start_linux.sh --verbose
|
RUN GPU_CHOICE=B LAUNCH_AFTER_INSTALL=FALSE INSTALL_EXTENSIONS=TRUE ./start_linux.sh --verbose
|
||||||
COPY CMD_FLAGS.txt /home/app/text-generation-webui/
|
COPY /user_data/CMD_FLAGS.txt /home/app/text-generation-webui/user_data
|
||||||
EXPOSE ${CONTAINER_PORT:-7860} ${CONTAINER_API_PORT:-5000} ${CONTAINER_API_STREAM_PORT:-5005}
|
EXPOSE ${CONTAINER_PORT:-7860} ${CONTAINER_API_PORT:-5000} ${CONTAINER_API_STREAM_PORT:-5005}
|
||||||
WORKDIR /home/app/text-generation-webui
|
WORKDIR /home/app/text-generation-webui
|
||||||
# set umask to ensure group read / write at runtime
|
# set umask to ensure group read / write at runtime
|
||||||
|
|
|
@ -41,14 +41,4 @@ services:
|
||||||
security_opt:
|
security_opt:
|
||||||
- seccomp=unconfined
|
- seccomp=unconfined
|
||||||
volumes:
|
volumes:
|
||||||
- ./cache:/home/app/text-generation-webui/cache
|
- ./user_data:/home/app/text-generation-webui/user_data
|
||||||
- ./characters:/home/app/text-generation-webui/characters
|
|
||||||
- ./extensions:/home/app/text-generation-webui/extensions
|
|
||||||
- ./loras:/home/app/text-generation-webui/loras
|
|
||||||
- ./logs:/home/app/text-generation-webui/logs
|
|
||||||
- ./models:/home/app/text-generation-webui/models
|
|
||||||
- ./presets:/home/app/text-generation-webui/presets
|
|
||||||
- ./prompts:/home/app/text-generation-webui/prompts
|
|
||||||
- ./softprompts:/home/app/text-generation-webui/softprompts
|
|
||||||
- ./training:/home/app/text-generation-webui/training
|
|
||||||
- ./cloudflared:/etc/cloudflared
|
|
||||||
|
|
|
@ -14,7 +14,7 @@ WORKDIR /home/app/
|
||||||
RUN git clone https://github.com/oobabooga/text-generation-webui.git
|
RUN git clone https://github.com/oobabooga/text-generation-webui.git
|
||||||
WORKDIR /home/app/text-generation-webui
|
WORKDIR /home/app/text-generation-webui
|
||||||
RUN GPU_CHOICE=D LAUNCH_AFTER_INSTALL=FALSE INSTALL_EXTENSIONS=TRUE ./start_linux.sh --verbose
|
RUN GPU_CHOICE=D LAUNCH_AFTER_INSTALL=FALSE INSTALL_EXTENSIONS=TRUE ./start_linux.sh --verbose
|
||||||
COPY CMD_FLAGS.txt /home/app/text-generation-webui/
|
COPY /user_data/CMD_FLAGS.txt /home/app/text-generation-webui/user_data
|
||||||
EXPOSE ${CONTAINER_PORT:-7860} ${CONTAINER_API_PORT:-5000} ${CONTAINER_API_STREAM_PORT:-5005}
|
EXPOSE ${CONTAINER_PORT:-7860} ${CONTAINER_API_PORT:-5000} ${CONTAINER_API_STREAM_PORT:-5005}
|
||||||
# set umask to ensure group read / write at runtime
|
# set umask to ensure group read / write at runtime
|
||||||
WORKDIR /home/app/text-generation-webui
|
WORKDIR /home/app/text-generation-webui
|
||||||
|
|
|
@ -41,12 +41,4 @@ services:
|
||||||
security_opt:
|
security_opt:
|
||||||
- seccomp=unconfined
|
- seccomp=unconfined
|
||||||
volumes:
|
volumes:
|
||||||
- ./characters:/home/app/text-generation-webui/characters
|
- ./user_data:/home/app/text-generation-webui/user_data
|
||||||
- ./extensions:/home/app/text-generation-webui/extensions
|
|
||||||
- ./loras:/home/app/text-generation-webui/loras
|
|
||||||
- ./models:/home/app/text-generation-webui/models
|
|
||||||
- ./presets:/home/app/text-generation-webui/presets
|
|
||||||
- ./prompts:/home/app/text-generation-webui/prompts
|
|
||||||
- ./softprompts:/home/app/text-generation-webui/softprompts
|
|
||||||
- ./training:/home/app/text-generation-webui/training
|
|
||||||
- ./cloudflared:/etc/cloudflared
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue