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-Chen 2024-03-14 20:04:01 +11:00
commit 8fa0aab61c
1 changed files with 29 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

35
pyproject.toml
View file

@ -32,18 +32,25 @@ include-package-data = false
exclude = ["images*"]
[project.optional-dependencies]
huggingface = [
huggingface-no-deps = [
"tyro",
"transformers>=4.38.2",
"datasets>=2.16.0",
"sentencepiece",
"accelerate>=0.26.1",
"trl>=0.7.9",
"peft>=0.7.1",
"tqdm",
"psutil",
"wheel>=0.42.0",
"numpy",
]
huggingface-deps =[
"accelerate>=0.26.1",
"trl>=0.7.9",
"peft>=0.7.1",
]
huggingface = [
"unsloth[huggingface-no-deps]",
"unsloth[huggingface-deps"],
]
cu118only = [
"xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp39-cp39-manylinux2014_x86_64.whl ; python_version=='3.9'",
"xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp310-cp310-manylinux2014_x86_64.whl ; python_version=='3.10'",
@ -160,8 +167,24 @@ colab-ampere-torch220 = [
"ninja",
"flash-attn",
]
colab = [
"unsloth[colab-ampere-torch220]",
colab-1 = [
"unsloth[huggingface-no-deps]",
]
colab-2 = [
"xformers",
"bitsandbytes",
"unsloth[huggingface-deps]",
]
colab-ampere-1 = [
"unsloth[huggingface-no-deps]",
]
colab-ampere-2 = [
"xformers",
"bitsandbytes",
"packaging",
"ninja",
"flash-attn",
"unsloth[huggingface-deps]",
]
colab-ampere = [
"unsloth[colab-ampere-torch220]",