diff --git a/requirements.txt b/requirements.txt index 30b822489f..b0edbae9d0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch==2.1.2 +torch==2.6.0 torchvision==0.16.2 torchaudio==2.1.2 xformers==0.0.23.post1 diff --git a/requirements_api.txt b/requirements_api.txt index fbb0035743..c3e8dfcef0 100644 --- a/requirements_api.txt +++ b/requirements_api.txt @@ -1,4 +1,4 @@ -torch~=2.1.2 +torch~=2.6.0 torchvision~=0.16.2 torchaudio~=2.1.2 xformers>=0.0.23.post1