78 lines
2.2 KiB
TOML
78 lines
2.2 KiB
TOML
# -------------------------------
|
|
# 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 = "zhangchi.usc1992@bytedance.com" },
|
|
{ name = "Bytedance - Seed - MLSys", email = "gmsheng@connect.hku.hk" },
|
|
]
|
|
|
|
# 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",
|
|
]
|
|
|
|
# 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"
|
|
] |