diff --git a/requirements.txt b/requirements.txt index 9b232d55..e7495ed4 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch>=2.2.0,<2.4.0 +torch>=2.2.0,<2.5.0 einops>=0.7.0,<0.8.0 memory-profiler bitsandbytes>=0.43.1,<0.44.0