diff --git a/pyproject.toml b/pyproject.toml index 6866317fb0..f604e6a809 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -204,6 +204,16 @@ cu126onlytorch260 = [ "xformers @ https://download.pytorch.org/whl/cu126/xformers-0.0.29.post3-cp311-cp311-win_amd64.whl ; python_version=='3.11' and platform_system == 'Windows'", "xformers @ https://download.pytorch.org/whl/cu126/xformers-0.0.29.post3-cp312-cp312-win_amd64.whl ; python_version=='3.12' and platform_system == 'Windows'", ] +cu128onlytorch270 = [ + "xformers @ https://download.pytorch.org/whl/cu128/xformers-0.0.30.post3-cp39-cp39-manylinux_2_28_x86_64.whl ; python_version=='3.9' and platform_system == 'Linux'", + "xformers @ https://download.pytorch.org/whl/cu128/xformers-0.0.30.post3-cp310-cp310-manylinux_2_28_x86_64.whl ; python_version=='3.10' and platform_system == 'Linux'", + "xformers @ https://download.pytorch.org/whl/cu128/xformers-0.0.30.post3-cp311-cp311-manylinux_2_28_x86_64.whl ; python_version=='3.11' and platform_system == 'Linux'", + "xformers @ https://download.pytorch.org/whl/cu128/xformers-0.0.30.post3-cp312-cp312-manylinux_2_28_x86_64.whl ; python_version=='3.12' and platform_system == 'Linux'", + "xformers @ https://download.pytorch.org/whl/cu128/xformers-0.0.30.post3-cp39-cp39-win_amd64.whl ; python_version=='3.9' and platform_system == 'Windows'", + "xformers @ https://download.pytorch.org/whl/cu128/xformers-0.0.30.post3-cp310-cp310-win_amd64.whl ; python_version=='3.10' and platform_system == 'Windows'", + "xformers @ https://download.pytorch.org/whl/cu128/xformers-0.0.30.post3-cp311-cp311-win_amd64.whl ; python_version=='3.11' and platform_system == 'Windows'", + "xformers @ https://download.pytorch.org/whl/cu128/xformers-0.0.30.post3-cp312-cp312-win_amd64.whl ; python_version=='3.12' and platform_system == 'Windows'", +] cu118 = [ "unsloth[huggingface]", "bitsandbytes>=0.43.3", @@ -314,6 +324,11 @@ cu126-torch260 = [ "bitsandbytes>=0.45.1", "unsloth[cu126onlytorch260]", ] +cu128-torch270 = [ + "unsloth[huggingface]", + "bitsandbytes>=0.45.5", + "unsloth[cu128onlytorch270]", +] kaggle = [ "unsloth[huggingface]", ] @@ -507,8 +522,14 @@ cu126-ampere-torch260 = [ "unsloth[cu126onlytorch260]", "unsloth[flashattention]", ] +cu128-ampere-torch270 = [ + "unsloth[huggingface]", + "bitsandbytes>=0.45.5", + "unsloth[cu128onlytorch270]", + "unsloth[flashattention]", +] [project.urls] homepage = "http://www.unsloth.ai" documentation = "https://github.com/unslothai/unsloth" -repository = "https://github.com/unslothai/unsloth" \ No newline at end of file +repository = "https://github.com/unslothai/unsloth"