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 16:51:13 +11:00
parent fdc7b14c4b
commit df2da4f043
1 changed files with 6 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

18
pyproject.toml
View file

@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
[project]
name = "unsloth"
version = "2023.11"
dynamic = ["version"]
description = "2X faster LLM finetuning"
readme = "README.md"
requires-python = ">=3.9"
@ -30,20 +30,14 @@ dependencies = [
"accelerate",
"trl",
"peft",
"torch>=2.1.0",
]
[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' and platform_system=='Linux'",
"xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp310-cp310-manylinux2014_x86_64.whl ; python_version=='3.10' and platform_system=='Linux'",
"xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp311-cp311-manylinux2014_x86_64.whl ; python_version=='3.11' and platform_system=='Linux'",
]
cu121 = [
"xformers @ https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp39-cp39-manylinux2014_x86_64.whl ; python_version=='3.9' and platform_system=='Linux'",
"xformers @ https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp310-cp310-manylinux2014_x86_64.whl ; python_version=='3.10' and platform_system=='Linux'",
"xformers @ https://download.pytorch.org/whl/cu121/xformers-0.0.22.post7-cp311-cp311-manylinux2014_x86_64.whl ; python_version=='3.11' and platform_system=='Linux'",
]
cu118 = {file = ["requirements_cu118.txt"]}
cu118 = {file = ["requirements_cu121.txt"]}
[project.urls]
homepage = "http://www.unsloth.ai"