Hugging Face's logo

Spaces:
vishaljoshi24
/
trl-4-dnd
Paused

App Files Community
trl-4-dnd
File size: 2,157 Bytes 
a080fe0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 
[metadata]
name = trl
version = 0.22.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 = [email protected]
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.55.0

[options.packages.find]
exclude =
    tests*

[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.9
peft =
    peft>=0.8.0
quantization =
    bitsandbytes
scikit =
    scikit-learn
test =
    parameterized
    pytest-cov
    pytest-rerunfailures
    pytest-xdist
    pytest
vllm =
    # vLLM package does not yet support Python 3.13. These constraints can be lifted once support is added:
    # see https://github.com/vllm-project/vllm/pull/13164
    vllm>=0.10.0; python_version < "3.13"
    fastapi; python_version < "3.13"
    pydantic; python_version < "3.13"
    requests; python_version < "3.13"
    uvicorn; python_version < "3.13"

vlm =
    Pillow
    torchvision
    num2words
dev =
    %(bco)s
    %(deepspeed)s
    %(diffusers)s
    %(judges)s
    %(liger)s
    %(peft)s
    %(quantization)s
    %(scikit)s
    %(test)s
    %(vlm)s

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

[coverage:run]
branch = True