diff --git a/constants.py b/constants.py
index d7a6d04..0fef791 100644
--- a/constants.py
+++ b/constants.py
@@ -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,
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 0f98426..c9828f2 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -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 . .
diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 18c6694..c29be6b 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -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