Files
SillyTavern-extras/docker/docker-compose.yml
2023-05-28 09:10:08 -05:00

20 lines
656 B
YAML

version: "3"
services:
sillytavern-extras:
runtime: nvidia
image: cohee1207/sillytavern-extras
build:
context: ../
dockerfile: docker/Dockerfile
args:
REQUIREMENTS: requirements.txt
MODULES: caption,summarize,classify
# REQUIREMENTS: requirements-complete.txt
# MODULES: caption,summarize,classify,keywords,prompt,sd,tts,chromadb
ports:
- "5100:5100"
environment:
- NVIDIA_VISIBLE_DEVICES=all
command: python server.py --enable-modules=caption,summarize,classify
# command: python server.py --enable-modules=caption,summarize,classify,keywords,prompt,sd,tts,chromadb