diff --git a/requirements.txt b/requirements.txt index 59274655..260a7663 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,5 +12,6 @@ easydict ftfy dashscope imageio-ffmpeg +psutil flash_attn -numpy>=1.23.5,<2 \ No newline at end of file +numpy>=1.23.5,<2