update deepspeech requirements

twosixlabs · Aug 8, 2023 · cf2943a · cf2943a
1 parent 3d82490
commit cf2943a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/docker/Dockerfile-pytorch-deepspeech b/docker/Dockerfile-pytorch-deepspeech
@@ -8,9 +8,7 @@ WORKDIR /armory-repo
 #       in the root of the repo.
 COPY ./ /armory-repo
 
-RUN echo "Installing Hydra" && \
-    pip install hydra-core && \
-    pip install git+https://github.com/romesco/hydra-lightning/\#subdirectory=hydra-configs-pytorch-lightning
+RUN pip install git+https://github.com/romesco/hydra-lightning/\#subdirectory=hydra-configs-pytorch-lightning
 
 RUN echo "Building Armory from local source"                                            && \
     echo "Updating Base Image..."                                                       && \

diff --git a/pyproject.toml b/pyproject.toml
@@ -77,7 +77,7 @@ engine = [
 
 pytorch = [
     "armory-testbed[engine,datasets,math]",
-    "torch == 1.12", # Pytorch 2 will break parts of armory
+    "torch >=1.12.0,<=1.13.1", # Pytorch 2 will break parts of armory
     "torchaudio",
     "torchvision",
 ]
@@ -98,6 +98,7 @@ deepspeech = [
     "google-cloud-storage",
     "transformers",
     "pytorch-lightning < 1.5.0",
+    "hydra-core",
 ]
 
 math = [