allow newer deps
Browse files- requirements.txt +3 -3
requirements.txt
CHANGED
|
@@ -4,9 +4,9 @@ bitsandbytes>=0.41.1
|
|
| 4 |
accelerate @ git+https://github.com/huggingface/accelerate@2a289f6108e77a77a4efffb3f6316bc98538413b
|
| 5 |
addict
|
| 6 |
fire
|
| 7 |
-
PyYAML
|
| 8 |
datasets
|
| 9 |
-
flash-attn
|
| 10 |
sentencepiece
|
| 11 |
wandb
|
| 12 |
einops
|
|
@@ -15,7 +15,7 @@ optimum
|
|
| 15 |
hf_transfer
|
| 16 |
colorama
|
| 17 |
numba
|
| 18 |
-
numpy
|
| 19 |
# qlora things
|
| 20 |
bert-score==0.3.13
|
| 21 |
evaluate==0.4.0
|
|
|
|
| 4 |
accelerate @ git+https://github.com/huggingface/accelerate@2a289f6108e77a77a4efffb3f6316bc98538413b
|
| 5 |
addict
|
| 6 |
fire
|
| 7 |
+
PyYAML>=6.0
|
| 8 |
datasets
|
| 9 |
+
flash-attn>=2.0.8
|
| 10 |
sentencepiece
|
| 11 |
wandb
|
| 12 |
einops
|
|
|
|
| 15 |
hf_transfer
|
| 16 |
colorama
|
| 17 |
numba
|
| 18 |
+
numpy>=1.24.4
|
| 19 |
# qlora things
|
| 20 |
bert-score==0.3.13
|
| 21 |
evaluate==0.4.0
|