LLM/airllm
1
0
Fork 0
mirror of https://github.com/0xSojalSec/airllm.git synced 2026-05-13 23:53:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

should not use head to avoid breaking changes, fix #11

Browse Source
This commit is contained in:
Yu Li
2023-07-03 07:29:37 +08:00
parent af775f264e
commit dc691b2958
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@@ -1,7 +1,7 @@
bitsandbytes==0.39.0
transformers @ git+https://github.com/huggingface/transformers.git
peft @ git+https://github.com/huggingface/peft.git
accelerate @ git+https://github.com/huggingface/accelerate.git
accelerate @ git+https://github.com/huggingface/accelerate.git@v0.20.3
einops==0.6.1
evaluate==0.4.0
scikit-learn==1.2.2