-
Notifications
You must be signed in to change notification settings - Fork 77
/
Copy pathpyproject.toml
79 lines (69 loc) · 2.18 KB
/
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
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
# -------------------------------
# build-system
# -------------------------------
[build-system]
requires = [
"setuptools>=61.0",
"wheel"
]
build-backend = "setuptools.build_meta"
# -------------------------------
# project (PEP 621 metadata)
# -------------------------------
[project]
name = "verl"
# We'll mark the version as "dynamic" because it's read from the file "verl/version/version"
# (PEP 621 calls this "dynamic version").
# The actual version is specified in the [tool.setuptools.dynamic] section below.
dynamic = ["version"]
description = "veRL: Volcano Engine Reinforcement Learning for LLM"
license = {file = "LICENSE"} # or "Apache-2.0", if you prefer an SPDX identifier
readme = {file = "README.md", content-type = "text/markdown"}
requires-python = ">=3.8"
authors = [
{ name = "Bytedance - Seed - MLSys", email = "[email protected]" },
{ name = "Bytedance - Seed - MLSys", email = "[email protected]" },
]
# Dependencies corresponding to install_requires in setup.py
dependencies = [
"accelerate",
"codetiming",
"datasets",
"dill",
"hydra-core",
"numpy",
"pybind11",
"ray",
"tensordict",
"transformers<4.48",
"vllm<=0.6.3",
"peft",
]
# Optional dependencies (extras_require in setup.py)
[project.optional-dependencies]
test = [
"pytest", "yapf"
]
# URLs
[project.urls]
Homepage = "https://github.com/volcengine/verl"
# -------------------------------
# tool.setuptools - Additional config
# -------------------------------
[tool.setuptools]
# True means `setuptools` will attempt to include all relevant files in package_data automatically.
# This corresponds to `include_package_data=True` in setup.py.
include-package-data = true
# We read the version from a file in 'verl/version/version'
[tool.setuptools.dynamic]
version = {file = "verl/version/version"}
# If you need to mimic `package_dir={'': '.'}`:
[tool.setuptools.package-dir]
"" = "."
# If you need to include specific non-Python data (like YAML files or version file):
# This is the rough equivalent of package_data={'': ['version/*'], 'verl': ['trainer/config/*.yaml']}
[tool.setuptools.package-data]
verl = [
"version/*",
"trainer/config/*.yaml"
]