diff --git a/requirements.txt b/requirements.txt index b24cb855b..6bbd04536 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,3 +14,4 @@ imageio-ffmpeg flash_attn gradio>=5.0.0 numpy==1.24.4 +einops