diff --git a/requirements.txt b/requirements.txt index b84828b..51c2340 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ timm==0.6.13 sentencepiece>=0.1.97 transformers>=4.34.1 +accelerate>=0.23.0