mergekit

1
[build-system]
2
requires = ["setuptools"]
3
build-backend = "setuptools.build_meta"
4

5
[project]
6
name = "mergekit"
7
description = "Tools for merging pre-trained large language models"
8
readme = "README.md"
9
license = { text = "LGPL-3.0-or-later" }
10
version = "0.0.4.1"
11
authors = [{ name = "Charles Goddard", email = "chargoddard@gmail.com" }]
12
dependencies = [
13
    "torch>=2.0.0",
14
    "tqdm==4.66.2",
15
    "click==8.1.7",
16
    "safetensors~=0.4.2",
17
    "accelerate~=0.27.2",
18
    "pydantic==2.6.2",
19
    "immutables==0.20",
20
    "transformers",
21
    "huggingface_hub",
22
    "peft",
23
    "typing-extensions",
24
    "sentencepiece",
25
    "protobuf",
26
]
27

28
[project.optional-dependencies]
29
dev = ["black~=24.2.0", "isort~=5.13.2", "pre-commit~=3.6.2"]
30
test = ["pytest~=8.0.1"]
31

32
[project.urls]
33
repository = "https://github.com/cg123/mergekit"
34

35

36
[project.scripts]
37
mergekit-yaml = "mergekit.scripts.run_yaml:main"
38
mergekit-mega = "mergekit.scripts.megamerge:main"
39
mergekit-legacy = "mergekit.scripts.legacy:main"
40
mergekit-layershuffle = "mergekit.scripts.layershuffle:main"
41
bakllama = "mergekit.scripts.bakllama:main"
42
mergekit-moe = "mergekit.scripts.mixtral_moe:main"
43

44
[tool.setuptools]
45
packages = [
46
    "mergekit",
47
    "mergekit.io",
48
    "mergekit.merge_methods",
49
    "mergekit.scripts",
50
    "mergekit._data",
51
    "mergekit._data.architectures",
52
]
53

54
[tool.isort]
55
profile = "black"
56

57
[tool.black]
58
line-length = 88
59
target-version = ['py37']
60
include = '\.pyi?$'
61

62
[tool.pytest.ini_options]
63
minversion = "6.0"
64
filterwarnings = [
65
    "ignore::pydantic.PydanticDeprecatedSince20:huggingface_hub.*:",
66
]
67
testpaths = ["tests"]
68