diff --git a/requirements.txt b/requirements.txt index 4644779..3b365c6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ ai2-tango[torch,transformers,fairscale,beaker,wandb]>=1.1 # Uncomment the following if you do. #beaker-py @ git+https://github.com/allenai/beaker-py.git@main -torchmetrics==0.11.1 +torchmetrics==1.0.1 more_itertools spacy>=3.0.0 wget