diff --git a/requirements.txt b/requirements.txt index 7aad4ff9b..85807cd38 100644 --- a/requirements.txt +++ b/requirements.txt @@ -67,5 +67,5 @@ axolotl-contribs-lgpl==0.0.6 axolotl-contribs-mit==0.0.3 # for sequence parallelism -yunchang.=0.6.0 +yunchang==0.6.0 ring-flash-attn>=0.1.4