SillyTavern/SillyTavern-extras
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Extras.git synced 2026-01-26 17:20:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

update docker setting

Browse Source
This commit is contained in:
donlinglok
2023-08-29 11:48:04 +08:00
parent 21639f1cd2
commit aa708fb574
3 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
constants.py
View File

@@ -13,7 +13,6 @@ DEFAULT_REMOTE_SD_PORT = 7860
DEFAULT_CHROMA_PORT = 8000
SILERO_SAMPLES_PATH = "tts_samples"
SILERO_SAMPLE_TEXT = "The quick brown fox jumps over the lazy dog"
# ALL_MODULES = ['caption', 'summarize', 'classify', 'keywords', 'prompt', 'sd']
DEFAULT_SUMMARIZE_PARAMS = {
"temperature": 1.0,
"repetition_penalty": 1.0,

2
docker/Dockerfile
View File

@@ -23,7 +23,7 @@ RUN conda create -n extras
RUN /bin/bash -c "source activate extras"
RUN conda install pytorch=2.0.0 torchvision=0.15.0 torchaudio=2.0.0 pytorch-cuda=11.7 git -c pytorch -c nvidia
RUN conda install pytorch torchvision torchaudio pytorch-cuda=11.7 git -c pytorch -c nvidia -c conda-forge
WORKDIR /sillytavern-extras/
COPY . .

4
docker/docker-compose.yml
View File

@@ -10,10 +10,10 @@ services:
REQUIREMENTS: requirements.txt
MODULES: caption,summarize,classify
# REQUIREMENTS: requirements-complete.txt
# MODULES: caption,summarize,classify,keywords,prompt,sd,tts,chromadb
# MODULES: caption,summarize,classify,sd,silero-tts,edge-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
# command: python server.py --enable-modules=caption,summarize,classify,sd,silero-tts,edge-tts,chromadb