Skip to content

Instantly share code, notes, and snippets.

@ksullivan86
Created April 1, 2024 19:17
Show Gist options
  • Save ksullivan86/2d6f43b9341a77f87594c1be3532f929 to your computer and use it in GitHub Desktop.
Save ksullivan86/2d6f43b9341a77f87594c1be3532f929 to your computer and use it in GitHub Desktop.
version: "3"
services:
text-generation-webui:
image: atinoda/text-generation-webui:default-arc # Specify variant as the :tag
container_name: text-generation-webui-arc
network_mode: docker_network
environment:
- TZ=America/Los_Angeles
- EXTRA_LAUNCH_ARGS="--listen --verbose" # Custom launch args (e.g., --model MODEL_NAME)
# - BUILD_EXTENSIONS_LIVE="silero_tts whisper_stt" # Install named extensions during every container launch. THIS WILL SIGNIFICANLTLY SLOW LAUNCH TIME.
- PUID=99
- PGID=100
ports:
- 7867:7860 # Default web port
- 5200:5000 # Default API port
- 5205:5005 # Default streaming port
- 5201:5001 # Default OpenAI API extension port
# labels:
# traefik.enable: true # allows traefik reverse proxy to see the app
# traefik.http.routers.text-generation-webui-cpu.entryPoints: https
# traefik.http.services.text-generation-webui-cpu.loadbalancer.server.port: 7860 # sepecifies port for traefik to route
volumes:
- /mnt/user/ai/text_generation_webui/models:/app/models
- /mnt/user/ai/appdata/text_generation_webui/text-generation-webui-arc/characters:/app/characters
- /mnt/user/ai/appdata/text_generation_webui/text-generation-webui-arc/loras:/app/loras
- /mnt/user/ai/appdata/text_generation_webui/text-generation-webui-arc/presets:/app/presets
- /mnt/user/ai/appdata/text_generation_webui/text-generation-webui-arc/prompts:/app/prompts
- /mnt/user/ai/appdata/text_generation_webui/text-generation-webui-arc/training:/app/training
- /mnt/user/ai/appdata/text_generation_webui/text-generation-webui-arc/extensions:/app/extensions # Persist all extensions
# - ./config/extensions/silero_tts:/app/extensions/silero_tts # Persist a single extension
logging:
driver: json-file
options:
max-file: "3" # number of files or file count
max-size: '10m'
# deploy:
# resources:
# reservations:
# devices:
# - driver: nvidia
# device_ids: ['0']
# capabilities: [gpu]
devices:
- /dev/dri:/dev/dri
restart: unless-stopped
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment