tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

bump hf dependencies (#1823)

Browse Source 
* bump hf dependencies

* revert optimum version change

* don't bump tokenizers all the way to 0.20 yet since transformers doesn't support that
This commit is contained in:
Wing Lian
2024-08-11 16:27:41 -04:00
committed by GitHub
parent 0801f239cc
commit 1853d6021d
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
requirements.txt
View File

@@ -1,18 +1,18 @@
--extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/ --extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
packaging==23.2 packaging==23.2
peft==0.12.0 peft==0.12.0
transformers==4.43.4 transformers==4.44.0
tokenizers==0.19.1 tokenizers>=0.19.1
bitsandbytes==0.43.3 bitsandbytes==0.43.3
accelerate==0.32.0 accelerate==0.33.0
datasets==2.20.0
deepspeed==0.14.4 deepspeed==0.14.4
pydantic==2.6.3 pydantic==2.6.3
addict addict
fire fire
PyYAML>=6.0 PyYAML>=6.0
requests requests
datasets==2.19.1 flash-attn==2.6.3
flash-attn==2.6.2
sentencepiece sentencepiece
wandb wandb
einops einops
@@ -37,8 +37,8 @@ autoawq>=0.2.5
mamba-ssm==1.2.0.post1 mamba-ssm==1.2.0.post1
# remote filesystems # remote filesystems
s3fs s3fs>=2024.5.0
gcsfs gcsfs>=2024.5.0
# adlfs # adlfs
trl==0.9.6 trl==0.9.6