| 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,sd,silero-tts,edge-tts,chromadb | |
| volumes: | |
| #- "./chromadb:/chromadb" | |
| - "./cache:/root/.cache" | |
| - "./api_key.txt:/sillytavern-extras/api_key.txt:rw" | |
| 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,sd,silero-tts,edge-tts,chromadb | |