trl/setup.cfg at main · respailab/trl · GitHub

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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
[metadata]
name = trl
version = 0.18.0.dev0
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

[options]
packages = find:
python_requires = >=3.9
include_package_data = True
install_requires =
    accelerate>=0.34.0
    datasets>=3.0.0
    transformers>=4.46.0

[options.packages.find]
exclude =
    tests*

[options.package_data]
trl =
    templates/*.md
    accelerate_configs/*.yaml

[options.extras_require]
bco =
    scikit-learn
    joblib
deepspeed =
    deepspeed>=0.14.4
diffusers =
    diffusers>=0.18.0
judges =
    openai>=1.23.2
    llm-blender>=0.0.2
liger =
    liger-kernel>=0.5.8
mergekit =
    mergekit>=0.0.5.1
peft =
    peft>=0.8.0
quantization =
    bitsandbytes
scikit =
    scikit-learn
test =
    parameterized
    pytest-cov
    pytest-rerunfailures
    pytest-xdist
    pytest
vllm =
    vllm>=0.8.3
    fastapi
    pydantic
    requests
    uvicorn
vlm =
    Pillow
dev =
    %(bco)s
    %(deepspeed)s
    %(diffusers)s
    %(judges)s
    %(liger)s
    %(mergekit)s
    %(peft)s
    %(quantization)s
    %(scikit)s
    %(test)s
    %(vllm)s
    %(vlm)s

[options.entry_points]
console_scripts =
    trl = trl.cli:main

[coverage:run]
branch = True