mirror of
https://github.com/huggingface/trl.git
synced 2025-10-20 18:43:52 +08:00
94 lines
1.9 KiB
INI
94 lines
1.9 KiB
INI
[metadata]
|
|
name = trl
|
|
version = file: VERSION
|
|
description = Train transformer language models with reinforcement learning.
|
|
long_description = file: README.md
|
|
long_description_content_type = text/markdown
|
|
author = Leandro von Werra
|
|
author_email = leandro.vonwerra@gmail.com
|
|
url = https://github.com/huggingface/trl
|
|
keywords = transformers, huggingface, language modeling, post-training, rlhf, sft, dpo, grpo
|
|
license_file = LICENSE
|
|
classifiers =
|
|
Development Status :: 2 - Pre-Alpha
|
|
Intended Audience :: Developers
|
|
Intended Audience :: Science/Research
|
|
Natural Language :: English
|
|
Operating System :: OS Independent
|
|
Programming Language :: Python :: 3
|
|
Programming Language :: Python :: 3.9
|
|
Programming Language :: Python :: 3.10
|
|
Programming Language :: Python :: 3.11
|
|
Programming Language :: Python :: 3.12
|
|
Programming Language :: Python :: 3.13
|
|
|
|
[options]
|
|
packages = find_namespace:
|
|
python_requires = >=3.9
|
|
include_package_data = True
|
|
install_requires =
|
|
accelerate>=1.4.0
|
|
datasets>=3.0.0
|
|
transformers>=4.56.1
|
|
transformers!=4.57.0; python_version == "3.9"
|
|
|
|
[options.packages.find]
|
|
exclude =
|
|
tests*
|
|
|
|
[options.extras_require]
|
|
bco =
|
|
scikit-learn
|
|
joblib
|
|
deepspeed =
|
|
deepspeed>=0.14.4
|
|
judges =
|
|
openai>=1.23.2
|
|
llm-blender>=0.0.2
|
|
liger =
|
|
liger-kernel>=0.6.2
|
|
peft =
|
|
peft>=0.8.0
|
|
quality =
|
|
pre-commit
|
|
hf-doc-builder
|
|
quantization =
|
|
bitsandbytes
|
|
scikit =
|
|
scikit-learn
|
|
test =
|
|
parameterized
|
|
pytest-cov
|
|
pytest-rerunfailures==15.1
|
|
pytest-xdist
|
|
pytest
|
|
vllm =
|
|
vllm==0.10.2
|
|
fastapi
|
|
pydantic
|
|
requests
|
|
uvicorn
|
|
|
|
vlm =
|
|
Pillow
|
|
torchvision
|
|
num2words==0.5.14
|
|
dev =
|
|
%(bco)s
|
|
%(deepspeed)s
|
|
%(judges)s
|
|
%(liger)s
|
|
%(peft)s
|
|
%(quality)s
|
|
%(quantization)s
|
|
%(scikit)s
|
|
%(test)s
|
|
%(vlm)s
|
|
|
|
[options.entry_points]
|
|
console_scripts =
|
|
trl = trl.cli:main
|
|
|
|
[coverage:run]
|
|
branch = True
|