From f01b6eaf430974064df16453552d7679587036b8 Mon Sep 17 00:00:00 2001 From: Gabriele Sarti Date: Thu, 5 Oct 2023 12:19:19 +0200 Subject: [PATCH] Pin torch in requirements --- requirements-dev.txt | 2 +- requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements-dev.txt b/requirements-dev.txt index 524cef56..d483cdfb 100644 --- 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 index 50a50681..6c08e8e1 100644 --- 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"