diff --git a/requirements.txt b/requirements.txt index 69c3b30..71fd64a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,8 @@ diffusers>=0.30.1 accelerate>=0.30.0 einops packaging -sentencepiece +sentencepiece>=0.2.0 timm>=0.6.12 numpy<=1.26.4 -protobuf>=4.25.4 \ No newline at end of file +protobuf>=4.25.4 +transformers>=4.44.2 \ No newline at end of file