mirror of
https://github.com/SillyTavern/SillyTavern-Extras.git
synced 2026-04-24 00:18:58 +00:00
01960773a1
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.
52 lines
1.8 KiB
YAML
52 lines
1.8 KiB
YAML
version: "3"
|
|
services:
|
|
sillytavern-extras:
|
|
# runtime: nvidia
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: 1
|
|
# either specify below which, or above how many. not both.
|
|
# device_ids: ['1']
|
|
capabilities: [gpu]
|
|
image: cohee1207/sillytavern-extras
|
|
build:
|
|
context: ../
|
|
dockerfile: docker/Dockerfile
|
|
args:
|
|
REQUIREMENTS: requirements.txt
|
|
# 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:
|
|
- "./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
|
|
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.
|