diff --git a/official/requirements.txt b/official/requirements.txt
index 53c799ef8f5cda611c6039c212c15541050c4192..4e23e67242e362962070c4f9b169982f92ed16c9 100644
--- a/official/requirements.txt
+++ b/official/requirements.txt
@@ -9,7 +9,7 @@ psutil>=5.4.3
 py-cpuinfo>=3.3.0
 scipy>=0.19.1
 tensorflow-hub>=0.6.0
-tensorflow-model-optimization>=0.1.3
+tensorflow-model-optimization>=0.2.1
 tensorflow_datasets
 typing
 sentencepiece