vrr/unsloth
2
0
Fork 0
mirror of https://github.com/unslothai/unsloth.git synced 2026-07-09 15:58:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 29

Update pyproject.toml

Browse source
This commit is contained in:
Daniel Han-Chen 2023-11-30 20:23:48 +11:00
parent 60c3301de0
commit 04587cd56e
1 changed files with 17 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

37
pyproject.toml
View file

@ -26,41 +26,38 @@ classifiers = [
version = {attr = "unsloth.__version__"}
[project.optional-dependencies]
kaggle = [
huggingface = [
"transformers",
"datasets",
"sentencepiece",
"accelerate",
"trl",
"peft",
"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'",
"xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp311-cp311-manylinux2014_x86_64.whl ; python_version=='3.11'",
]
cu118 = [
"transformers",
"bitsandbytes",
"datasets",
"sentencepiece",
"accelerate",
"trl",
"peft",
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'",
"xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp311-cp311-manylinux2014_x86_64.whl ; python_version=='3.11'",
]
cu121 = [
"transformers",
"bitsandbytes",
"datasets",
"sentencepiece",
"accelerate",
"trl",
"peft",
cu121only = [
"xformers @ https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp39-cp39-manylinux2014_x86_64.whl ; python_version=='3.9'",
"xformers @ https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl ; python_version=='3.10'",
"xformers @ https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp311-cp311-manylinux2014_x86_64.whl ; python_version=='3.11'",
]
cu118 = [
"unsloth[huggingface]",
"bitsandbytes",
"unsloth[cu118only]",
]
cu121 = [
"unsloth[huggingface]",
"bitsandbytes",
"unsloth[cu121only]",
]
kaggle = [
"unsloth[huggingface]",
"unsloth[cu118only]",
]
[project.urls]
homepage = "http://www.unsloth.ai"