Merge pull request #128 from JSchmie/roll-back-torch-verision

align minimal torch version
JSchmie · Oct 24, 2024 · 9528468 · 9528468
2 parents 3fe1380 + 81fefd5
commit 9528468
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 260 deletions.
diff --git a/environment.yml b/environment.yml
diff --git a/pyproject.toml b/pyproject.toml
@@ -33,10 +33,10 @@ exclude =[
 python = "^3.9"
 tqdm = "^4.66.5"
 numpy = "^1.26.4"
-openai-whisper = "^20231117"
+openai-whisper = ">=20231117,<20240931"
 faster-whisper = "^1.0.3"
 "pyannote.audio" = "^3.3.1"
-torch = "^2.3.0"
+torch = "^2.1.2"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.1.1"
@@ -57,7 +57,7 @@ format-jinja = """
 
 [tool.poetry.group.docs.dependencies]
 sphinx = "^7.3.7"
-sphinx-rtd-theme = "^2.0.0"
+sphinx-rtd-theme = ">=2,<4"
 markdown-it-py = {version = "~3.0.0", extras = ["plugins"]}
 myst-parser = "^3.0.1"
 mdit-py-plugins = "^0.4.1"

diff --git a/requirements.txt b/requirements.txt
@@ -10,5 +10,5 @@ pyannote.database~=5.0.1
 pyannote.metrics~=3.2.1
 pyannote.pipeline~=3.0.1
 
-torch>=2.0.0
+torchaudio>=2.1.2