-
Notifications
You must be signed in to change notification settings - Fork 27
/
pyproject.toml
40 lines (35 loc) · 1014 Bytes
/
pyproject.toml
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
[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "mergoo"
version = "0.0.10"
description = "Impelementation of Leeroo LLM composer."
authors = [{ name = "Leeroo Team", email = "support@leeroo.com" }]
readme = "readme.md"
keywords = ["LLM", "compose", "MoE", "router", "mixture-of-adapters", "merge"]
license = {file = "LICENSE"}
dependencies = [
"torch>=2.0.0",
"tqdm==4.66.3",
"safetensors~=0.4.2",
"accelerate~=0.27.2",
"transformers",
"huggingface_hub",
"peft",
"typing-extensions",
"sentencepiece",
"protobuf",
"numpy",
]
[project.urls]
homepage = "https://github.com/Leeroo-AI/mergoo"
repository = "https://github.com/Leeroo-AI/mergoo"
[tool.mypy]
# https://mypy.readthedocs.io/en/latest/config_file.html#using-a-pyproject-toml-file
python_version = "3.10"
strict = false
[project.optional-dependencies]
dev = ["super-lint"]
[tool.setuptools]
packages = ["mergoo", "mergoo.composers", "mergoo.models"]