diff --git a/vilbert/vilbert-multi-task-master/requirements.txt b/vilbert/vilbert-multi-task-master/requirements.txt
index 1a3fa4a..1947b9b 100644
--- a/vilbert/vilbert-multi-task-master/requirements.txt
+++ b/vilbert/vilbert-multi-task-master/requirements.txt
@@ -6,7 +6,7 @@ tensorflow==1.15.2
 tensorpack==0.9.4
 tqdm==4.31.1
 easydict==1.9
-PyYAML==5.1.2
+PyYAML==5.4
 jsonlines==1.2.0
 json-lines==0.5.0
 matplotlib
diff --git a/vilbert/vilbert-multi-task-master/requirements_me2020.txt b/vilbert/vilbert-multi-task-master/requirements_me2020.txt
index e1c1fae..1f735d9 100644
--- a/vilbert/vilbert-multi-task-master/requirements_me2020.txt
+++ b/vilbert/vilbert-multi-task-master/requirements_me2020.txt
@@ -49,7 +49,7 @@ python-prctl==1.7
 pytorch-transformers==1.2.0
 pytz==2020.1
 PyWavelets==1.1.1
-PyYAML==5.1.2
+PyYAML==5.4
 pyzmq==19.0.2
 regex==2020.9.27
 requests==2.24.0