diff --git a/requirements.txt b/requirements.txt
index c41d17ffaf7..fb265a3e73b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -13,7 +13,7 @@ torchvision
 transformers>=4.31.0
 tokenizers>=0.13.3
 spacy>=2.3
-PyYAML>=3.12,<6.0.1
+PyYAML>=3.12,<6.0.1,!=5.4
 absl-py
 kaggle
 requests