From b2eb882bb2fb14c88a1628a47f6ccd6a20a7b89d Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Thu, 30 Nov 2023 16:54:10 +1100 Subject: [PATCH] Update pyproject.toml --- pyproject.toml | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index eb3fcde9bb..562ee206a7 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,14 +30,23 @@ dependencies = [ "accelerate", "trl", "peft", + "torch>=2.1.0", ] [tool.setuptools.dynamic] version = {attr = "unsloth.__version__"} [project.optional-dependencies] -cu118 = {file = ["requirements_cu118.txt"]} -cu121 = {file = ["requirements_cu121.txt"]} +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'", +] [project.urls] homepage = "http://www.unsloth.ai"