diff --git a/pyproject.toml b/pyproject.toml index d37fa60..20f1e68 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,7 +11,7 @@ authors = [ {name = "Your Name", email = "your.email@example.com"} ] dependencies = [ - "torch==2.1.1", + "torch==2.2.0", "pytorch-lightning==2.1.1", "transformers==4.45.2", "numpy==1.24.4", diff --git a/requirements.txt b/requirements.txt index 43147cf..7d69128 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch~=2.1.1 +torch~=2.2.0 pytorch-lightning~=2.1.1 transformers==4.45.2 numpy==1.24.4