diff --git a/src/requirements.txt b/src/requirements.txt index 4d5cb00..6171d89 100644 --- a/src/requirements.txt +++ b/src/requirements.txt @@ -1,7 +1,7 @@ tqdm sentencepiece datasets -transformers==4.31.0 +transformers==4.36.0 deepspeed==0.11.0 accelerate==0.21.0 flash_attn==2.3.6