diff --git a/scripts/colab_reqs/rome.txt b/scripts/colab_reqs/rome.txt index f6888821..66d39b76 100644 --- a/scripts/colab_reqs/rome.txt +++ b/scripts/colab_reqs/rome.txt @@ -1,3 +1,5 @@ datasets==1.18.3 python-dotenv==0.19.2 git+https://github.com/kmeng01/transformers-colab@allennlp-compat +transformers==4.33.1 +accelerate==0.23.0