diff --git a/requirements.txt b/requirements.txt index 35bb79c..87052af 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,7 +4,7 @@ nltk==3.6.6 pyLDAvis==2.1.2 spacy==3.0.5 tensorflow==2.6.3 -tensorflow-gpu==2.6.3 +tensorflow-gpu==2.12.0 torch==1.8.0 tqdm transformers==4.17.0