diff --git a/requirements.txt b/requirements.txt index 740de24..1943e9c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,4 +6,4 @@ numpy<2 rotary_embedding_torch scipy setuptools -torch==2.0.1 +torch==2.2.0