diff --git a/requirements.txt b/requirements.txt index b89b147..cd150c7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch==2.0.1 +torch==2.2.1 transformers==4.38.2 xformers>=0.0.21 timm>=0.9.16