Create docker-compose.yml

This commit is contained in:
ramblingcoder
2023-05-28 09:10:08 -05:00
committed by GitHub
parent 8d402fa02d
commit 630f9a0054

19
docker/docker-compose.yml Normal file
View File

@@ -0,0 +1,19 @@
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