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

Update deps 202401 (#1204) [skip ci]

Browse Source 
* update deps

* xformers fix too
This commit is contained in:
Wing Lian
2024-01-25 10:11:49 -05:00
committed by GitHub
parent 33e117088f
commit a01b998c0f
2 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
requirements.txt
View File

@@ -1,6 +1,6 @@
--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.7.0 peft==0.7.1
transformers==4.37.0 transformers==4.37.0
tokenizers==0.15.0 tokenizers==0.15.0
bitsandbytes>=0.41.1 bitsandbytes>=0.41.1
@@ -15,16 +15,14 @@ sentencepiece
wandb wandb
einops einops
xformers==0.0.22 xformers==0.0.22
optimum==1.13.2 optimum==1.16.2
hf_transfer hf_transfer
colorama colorama
numba numba
numpy>=1.24.4 numpy>=1.24.4
mlflow mlflow
# qlora things # qlora things
bert-score==0.3.13
evaluate==0.4.0 evaluate==0.4.0
rouge-score==0.1.2
scipy scipy
scikit-learn==1.2.2 scikit-learn==1.2.2
pynvml pynvml

4
setup.py
View File

@@ -27,9 +27,9 @@ def parse_requirements():
try: try:
torch_version = version("torch") torch_version = version("torch")
if torch_version.startswith("2.1.1"): if torch_version.startswith("2.1."):
_install_requires.pop(_install_requires.index("xformers==0.0.22")) _install_requires.pop(_install_requires.index("xformers==0.0.22"))
_install_requires.append("xformers==0.0.23") _install_requires.append("xformers>=0.0.23")
except PackageNotFoundError: except PackageNotFoundError:
pass pass