unsloth-zoo/pyproject.toml at fb0ba59f59a968547892ac049eb32bf2f7139cb2 · unslothai/unsloth-zoo · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
[build-system]
requires = ["setuptools==82.0.1", "setuptools-scm==9.2.2"]
build-backend = "setuptools.build_meta"

[project]
name = "unsloth_zoo"
dynamic = ["version"]
description = "Utils for Unsloth"
readme = "README.md"
requires-python = ">=3.9,<3.15"
license = "LGPL-3.0-or-later"
keywords = ["ai", "llm",]
authors = [
    {email = "info@unsloth.ai"},
    {name = "Unsloth AI team"},
]
maintainers = [
    {name = "Daniel Han", email = "danielhanchen@gmail.com"},
    {name = "Michael Han", email = "info@unsloth.ai"},
]
classifiers = [
    "Programming Language :: Python",
]
dependencies = [
    # --- GPU-only deps (skipped on macOS arm64) ---
    "torch>=2.4.0,<2.13.0 ; (sys_platform != 'darwin' or platform_machine != 'arm64')",
    "torchao>=0.13.0 ; (sys_platform != 'darwin' or platform_machine != 'arm64')",
    "triton>=3.0.0 ; ('linux' in sys_platform)",
    "tyro ; (sys_platform != 'darwin' or platform_machine != 'arm64')",
    "accelerate>=0.34.1 ; (sys_platform != 'darwin' or platform_machine != 'arm64')",
    "trl>=0.18.2,!=0.19.0,<=0.24.0 ; (sys_platform != 'darwin' or platform_machine != 'arm64')",
    "peft>=0.18.0,!=0.11.0 ; (sys_platform != 'darwin' or platform_machine != 'arm64')",
    "cut_cross_entropy ; python_version >= '3.10' and (sys_platform != 'darwin' or platform_machine != 'arm64')",
    # --- MLX-only deps (macOS arm64) ---
    "mlx>=0.22.0 ; (sys_platform == 'darwin' and platform_machine == 'arm64')",
    "mlx-lm>=0.22.0 ; (sys_platform == 'darwin' and platform_machine == 'arm64')",
    "mlx-vlm>=0.4.4 ; (sys_platform == 'darwin' and platform_machine == 'arm64')",
    # --- Shared deps (all platforms) ---
    "packaging>=24.1",
    "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.4,!=4.57.5,!=5.0.0,!=5.1.0,<=5.5.0",
    "datasets>=4.0,<4.6",
    "sentencepiece>=0.2.0",
    "tqdm",
    "psutil",
    "wheel>=0.42.0",
    "numpy",
    "protobuf",
    "huggingface_hub>=0.34.0",
    "hf_transfer",
    "pillow",
    "regex",
    "msgspec",
    "typing_extensions",
    "filelock",
]

[tool.setuptools.dynamic]
version = {attr = "unsloth_zoo.__init__.__version__"}

[tool.setuptools]
include-package-data = false

[tool.setuptools.package-data]
unsloth_zoo = ["diffusion_studio/*.html"]

[tool.setuptools.packages.find]
exclude = ["images*"]

[project.optional-dependencies]
mlx = [
    "mlx>=0.22.0",
    "mlx-lm>=0.22.0",
    "mlx-vlm>=0.4.4",
]
core = [
    "packaging>=24.1",
    "tyro",
    "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.4,!=4.57.5,!=5.0.0,!=5.1.0,<=5.5.0",
    "datasets>=4.0,<4.6",
    "sentencepiece>=0.2.0",
    "tqdm",
    "psutil",
    "wheel>=0.42.0",
    "numpy",
    "accelerate>=0.34.1",
    "trl>=0.18.2,!=0.19.0,<=0.24.0",
    "peft>=0.18.0,!=0.11.0",
    "protobuf",
    "huggingface_hub>=0.34.0",
    "hf_transfer",
    "cut_cross_entropy ; python_version >= '3.10'",
    "pillow",
    "regex",
    "msgspec",
    "typing_extensions",
    "filelock",
]
huggingface = [
    "unsloth_zoo[core]",
    "triton-windows ; (sys_platform == 'win32') and (platform_machine == 'AMD64' or platform_machine == 'x86_64')",
]
base = [
    "unsloth_zoo[huggingface]",
]
intelgpu = [
    "torch>=2.4.0",
    "triton>=3.0.0 ; ('linux' in sys_platform)",
    "unsloth_zoo[core]",
]

[project.urls]
homepage = "http://www.unsloth.ai"
documentation = "https://github.com/unslothai/unsloth"
repository = "https://github.com/unslothai/unsloth"

[tool.pytest.ini_options]
# Mirrors the same stanza added to unslothai/unsloth in PR #5397.
# Scopes `pytest` discovery to ./tests so the GPU-heavy tests there
# don't accidentally fire when a developer runs a bare `pytest` from
# the repo root. CI jobs explicitly name the target subtree (e.g.
# `pytest tests/security` for the hard-gate suite).
testpaths = ["tests"]
pythonpath = ["."]