Spaces:
Sleeping
Sleeping
version: "3.3" | |
services: | |
text-generation-webui: | |
build: | |
context: . | |
args: | |
# Requirements file to use: | |
# | GPU | requirements file to use | | |
# |--------|---------| | |
# | NVIDIA | `requirements.txt` | | |
# | AMD | `requirements_amd.txt` | | |
# | CPU only | `requirements_cpu_only.txt` | | |
# | Apple Intel | `requirements_apple_intel.txt` | | |
# | Apple Silicon | `requirements_apple_silicon.txt` | | |
# Default: requirements.txt` | |
# BUILD_REQUIREMENTS: requirements.txt | |
# Extension requirements to build: | |
# BUILD_EXTENSIONS: | |
# specify which cuda version your card supports: https://developer.nvidia.com/cuda-gpus | |
TORCH_CUDA_ARCH_LIST: ${TORCH_CUDA_ARCH_LIST:-7.5} | |
BUILD_EXTENSIONS: ${BUILD_EXTENSIONS:-} | |
APP_GID: ${APP_GID:-6972} | |
APP_UID: ${APP_UID-6972} | |
env_file: .env | |
user: "${APP_RUNTIME_UID:-6972}:${APP_RUNTIME_GID:-6972}" | |
ports: | |
- "${HOST_PORT:-7860}:${CONTAINER_PORT:-7860}" | |
- "${HOST_API_PORT:-5000}:${CONTAINER_API_PORT:-5000}" | |
stdin_open: true | |
group_add: | |
- video | |
tty: true | |
ipc: host | |
devices: | |
- /dev/kfd | |
- /dev/dri | |
cap_add: | |
- SYS_PTRACE | |
security_opt: | |
- seccomp=unconfined | |
volumes: | |
- ./characters:/home/app/text-generation-webui/characters | |
- ./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 | |