ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update pyproject.toml

Browse source
This commit is contained in:
Daniel Han 2024-07-18 10:59:09 -07:00
commit 7fb3838c00
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
pyproject.toml
View file

@ -46,6 +46,7 @@ huggingface = [
"trl>=0.7.9,<0.9.0",
"peft>=0.7.1,!=0.11.0",
"protobuf<4.0.0",
"huggingface_hub[hf_transfer]",
]
cu118only = [
"xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp39-cp39-manylinux2014_x86_64.whl ; python_version=='3.9'",
@ -195,6 +196,7 @@ colab-new = [
"wheel>=0.42.0",
"numpy",
"protobuf<4.0.0",
"huggingface_hub[hf_transfer]",
]
colab-no-deps = [
"accelerate>=0.26.1",