diff --git a/requirements.txt b/requirements.txt
index eef377f..665602e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -17,4 +17,4 @@ torchaudio
 modelscope
 sentencepiece
 transformers
-yaml
\ No newline at end of file
+PyYAML
\ No newline at end of file