allow newer deps

This commit is contained in:
Aman Karmani
2023-08-26 15:06:05 +00:00
parent 0b7ba57ec4
commit c29117a0d7

View File

@@ -4,9 +4,9 @@ bitsandbytes>=0.41.1
accelerate @ git+https://github.com/huggingface/accelerate@2a289f6108e77a77a4efffb3f6316bc98538413b accelerate @ git+https://github.com/huggingface/accelerate@2a289f6108e77a77a4efffb3f6316bc98538413b
addict addict
fire fire
PyYAML==6.0 PyYAML>=6.0
datasets datasets
flash-attn==2.0.8 flash-attn>=2.0.8
sentencepiece sentencepiece
wandb wandb
einops einops
@@ -15,7 +15,7 @@ optimum
hf_transfer hf_transfer
colorama colorama
numba numba
numpy==1.24.4 numpy>=1.24.4
# qlora things # qlora things
bert-score==0.3.13 bert-score==0.3.13
evaluate==0.4.0 evaluate==0.4.0