diff --git a/requirements.txt b/requirements.txt index f707946a02..4d74dee908 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,7 +11,7 @@ addict fire PyYAML>=6.0 requests -datasets>=2.15.0 +datasets==2.15.0 flash-attn==2.5.5 sentencepiece wandb