DistillKit/pyproject.toml at main · arcee-ai/DistillKit · 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
[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"

[project]
name = "distillkit"
description = "Tools for online and offline distillation of large language models."
version = "1.0.0"
authors = [
    { name = "Charles Goddard", email = "charles@arcee.ai" },
    { name = "Lucas Atkins", email = "lucas@arcee.ai" },
]
requires-python = ">=3.10"
dependencies = [
    "trl~=0.25.1",
    "accelerate~=1.11.0",
    "transformers>=4.50.3",
    "datasets>=3.5.0",
    "pydantic~=2.12.4",
    "click~=8.3.0",
    "torch>=2.0.0",
    "wandb",
    "tqdm",
]


[project.scripts]
distillkit = "distillkit.main:main"

[project.optional-dependencies]
dev = ["pytest~=9.0.2", "ruff~=0.14.9"]
capture = [
    "vllm>=0.12.0",
    "pyarrow",
]

[project.urls]
repository = "https://github.com/arcee-ai/distillkit"

[tool.setuptools]
packages = ["distillkit", "distillkit.compression", "distillkit.lossfuncs"]