diff --git a/requirements.txt b/requirements.txt index b60c406..0a679d0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,7 +9,7 @@ optuna==2.10.0 img2vec_pytorch==0.2.5 pypinyin==0.44.0 keras==2.7.0 -tensorflow==2.7.0 +tensorflow==2.11.1 gensim==4.2.0 datasets transformers \ No newline at end of file diff --git a/setup.py b/setup.py index 9ea41d7..88902bc 100644 --- a/setup.py +++ b/setup.py @@ -22,7 +22,7 @@ 'img2vec_pytorch==0.2.5', 'pypinyin==0.44.0', 'keras==2.7.0', - 'tensorflow==2.7.0', + 'tensorflow==2.11.1', 'gensim==4.2.0', # 'glove-python-binary', # 'transformers',