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 2023-11-30 20:20:45 +11:00
commit 60c3301de0
1 changed files with 23 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

33
pyproject.toml
View file

@ -22,26 +22,41 @@ classifiers = [
"Programming Language :: Python",
]
dependencies = [
"transformers",
[tool.setuptools.dynamic]
version = {attr = "unsloth.__version__"}
[project.optional-dependencies]
kaggle = [
"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",
]
[tool.setuptools.dynamic]
version = {attr = "unsloth.__version__"}
[project.optional-dependencies]
cu118 = [
"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",
"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'",