diff --git a/lite/examples/model_personalization/converter/requirements.txt b/lite/examples/model_personalization/converter/requirements.txt index 5555fc69294..d0564ddae1a 100644 --- a/lite/examples/model_personalization/converter/requirements.txt +++ b/lite/examples/model_personalization/converter/requirements.txt @@ -1,3 +1,3 @@ -tensorflow==2.0.0rc0 +tensorflow==2.5.1 Pillow==6.2.0 scipy==1.3.0 diff --git a/lite/examples/model_personalization/converter/setup.py b/lite/examples/model_personalization/converter/setup.py index 105ac18b89e..86b046add7f 100644 --- a/lite/examples/model_personalization/converter/setup.py +++ b/lite/examples/model_personalization/converter/setup.py @@ -25,7 +25,7 @@ version='0.1', packages=find_packages(), install_requires=[ - 'tensorflow==2.0.0rc0', + 'tensorflow==2.5.1', 'Pillow>=6.1.0,<7.0', 'scipy>=1.3.0,<2.0', ],