diff --git a/requirements.txt b/requirements.txt index 7fea4309..2e09bcd1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch>=2.2.0,<2.3.0 +torch>=2.2.0,<2.4.0 einops>=0.7.0,<0.8.0 memory-profiler bitsandbytes>=0.41.3.post2,<0.42.0