diff --git a/requirements.txt b/requirements.txt index 61c87abb..8daede46 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ rouge_score fire openai transformers>=4.26.1 -torch +torch>=1.13.1 sentencepiece tokenizers==0.12.1 wandb