diff --git a/requirements.txt b/requirements.txt
index 9b78aad27..f464c9379 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,5 +8,6 @@ safetensors>=0.3.0
 pytorch_lightning
 aiohttp
 accelerate
+pyyaml
 ruamel.yaml
 pytest