SillyTavern/SillyTavern-extras
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern-Extras.git synced 2026-05-01 11:51:22 +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 DEFAULT_CHROMA_PORT = 8000
SILERO_SAMPLES_PATH = "tts_samples" SILERO_SAMPLES_PATH = "tts_samples"
SILERO_SAMPLE_TEXT = "The quick brown fox jumps over the lazy dog" SILERO_SAMPLE_TEXT = "The quick brown fox jumps over the lazy dog"
# ALL_MODULES = ['caption', 'summarize', 'classify', 'keywords', 'prompt', 'sd']
DEFAULT_SUMMARIZE_PARAMS = { DEFAULT_SUMMARIZE_PARAMS = {
"temperature": 1.0, "temperature": 1.0,
"repetition_penalty": 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 /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/ WORKDIR /sillytavern-extras/
COPY . . COPY . .

4
docker/docker-compose.yml
View File

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