diff --git a/pyproject.toml b/pyproject.toml index 337c56b6d..4e16ab067 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "unsloth" -dynamic = ["version", "dependencies", "optional-dependencies"] +dynamic = ["version"] description = "2X faster LLM finetuning" readme = "README.md" requires-python = ">=3.9" @@ -22,10 +22,30 @@ classifiers = [ "Programming Language :: Python", ] +dependencies = [ + "transformers", + "bitsandbytes", + "datasets", + "sentencepiece", + "accelerate", + "trl", + "peft", +] + [tool.setuptools.dynamic] version = {attr = "unsloth.__version__"} -dependencies = {file = ["requirements.txt"]} -optional-dependencies = {cu118 = { file = ["requirements_cu118.txt"] }} + +[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 = [ + "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'", +] [project.urls] homepage = "http://www.unsloth.ai"