diff --git a/model_hub/tests/requirements.txt b/model_hub/tests/requirements.txt
index 3ff2390791d..ccf997c6442 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.4.0
 
 # install mmcv and mmdet
 mmdet==2.21.0