diff --git a/model_hub/tests/requirements.txt b/model_hub/tests/requirements.txt
index 3ff2390791d..1f9fccaef91 100644
--- a/model_hub/tests/requirements.txt
+++ b/model_hub/tests/requirements.txt
@@ -5,7 +5,7 @@ coverage
 
 # install transformers and datasets
 transformers==4.8.2
-datasets==1.9.0
+datasets==2.3.2
 
 # install mmcv and mmdet
 mmdet==2.21.0