Pin torch in requirements

inseq-team · Oct 5, 2023 · f01b6ea · f01b6ea
1 parent 3461f0c
commit f01b6ea
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -144,7 +144,7 @@ sympy==1.12 ; python_full_version >= "3.8.1" and python_version < "3.12"
 threadpoolctl==3.2.0 ; python_full_version >= "3.8.1" and python_version < "3.12"
 tokenizers==0.13.3 ; python_full_version >= "3.8.1" and python_version < "3.12"
 tomli==2.0.1 ; python_full_version >= "3.8.1" and python_version < "3.12"
-torch==2.0.0 ; python_full_version >= "3.8.1" and python_version < "3.12"
+torch==2.1.0 ; python_full_version >= "3.8.1" and python_version < "3.12"
 torchtyping==0.1.4 ; python_full_version >= "3.8.1" and python_version < "3.12"
 tornado==6.3.3 ; python_full_version >= "3.8.1" and python_version < "3.12"
 tqdm==4.66.1 ; python_full_version >= "3.8.1" and python_version < "3.12"

diff --git a/requirements.txt b/requirements.txt
@@ -50,7 +50,7 @@ six==1.16.0 ; python_full_version >= "3.8.1" and python_version < "3.12"
 sympy==1.12 ; python_full_version >= "3.8.1" and python_version < "3.12"
 tokenizers==0.13.3 ; python_full_version >= "3.8.1" and python_version < "3.12"
 tomli==2.0.1 ; python_full_version >= "3.8.1" and python_version < "3.12"
-torch==2.0.0 ; python_full_version >= "3.8.1" and python_version < "3.12"
+torch==2.1.0 ; python_full_version >= "3.8.1" and python_version < "3.12"
 torchtyping==0.1.4 ; python_full_version >= "3.8.1" and python_version < "3.12"
 tqdm==4.66.1 ; python_full_version >= "3.8.1" and python_version < "3.12"
 transformers[sentencepiece,tokenizers,torch]==4.33.3 ; python_full_version >= "3.8.1" and python_version < "3.12"