diff --git a/requirements.txt b/requirements.txt index 0aa4567..25bcf16 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,4 +14,4 @@ fire tokenizers transformers bitsandbytes>=0.49.2 -accelerate>=0.24.0 +accelerate>=1.13.0 diff --git a/requirements_cpu.txt b/requirements_cpu.txt index 2b063ff..d572189 100644 --- a/requirements_cpu.txt +++ b/requirements_cpu.txt @@ -2,7 +2,7 @@ sentencepiece torch tokenizers==0.15.2 transformers==4.39.3 -accelerate>=0.24.0 +accelerate>=1.13.0 tqdm pandas matplotlib