Update docker-compose.yml

Added additional documentation for requirements, the different options, how to format the requirements, removed the MODULES argument, added commented lines for additional volume for the RVC module (Known to still be missing some. Experimentation required.), change the final "command" to "entrypoint" and add documentation.
This commit is contained in:
mrguymiah
2024-04-02 21:09:41 -04:00
committed by GitHub
parent c3529a5fed
commit 01960773a1

View File

@@ -17,16 +17,35 @@ services:
dockerfile: docker/Dockerfile
args:
REQUIREMENTS: requirements.txt
MODULES: caption,summarize,classify
# REQUIREMENTS: "requirements.txt -r requirements-rvc.txt"
# MODULES: caption,summarize,classify,rvc
# Possible Values:
# requirements.txt - Use when running on Nvidia
# requirements-rocm.txt - Use when running on AMD
# requirements-silicon.txt - Use for Apple M1 or M2 chipset.
#
# Add-on Instructions:
# Append the above with " -r " as separator and surround with quotes.
# Example: REQUIREMENTS: "requirements.txt -r requirements-rvc.txt"
#
# Possible Add-ons:
# requirements-coqui.txt - Required to run the COQUI TTS Module
# requirements-rvc.txt - Required to run the Real-time Voice Cloning module
#
volumes:
#- "./chromadb:/chromadb"
- "./cache:/root/.cache"
- "../api_key.txt:/sillytavern-extras/api_key.txt:rw"
# Options Module Volumes - Uncomment the lines for your desired modules before building.
#
# ChromaDB - Utilized by ChromaDB.
# - "./chromadb:/chromadb"
#
# Real-time Voice CLoning
# - "./models/rvc:/data/models/rvc"
#
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,rvc
entrypoint: ["python", "server.py", "--enable-modules=caption,summarize,classify"]
#
# For a list of possible modules, refer to the "Modules" section on the project's main page.
# NOTE - Separate additional modules by commas in the "--enable-modules" list. DO NOT separate them with spaces.