diff --git a/requirements.txt b/requirements.txt index 75f0913..9a28684 100644 --- a/requirements.txt +++ b/requirements.txt @@ -13,7 +13,7 @@ transformers openai # Machine Learning packages -torch == 2.1.0 +torch == 2.2.0 triton == 2.1.0 xformers == 0.0.22.post7