forked from CarperAI/trlx
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.cfg
72 lines (68 loc) · 2.03 KB
/
setup.cfg
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
[metadata]
name = trlx
author = Alex Havrilla
version = 0.6.0
url = https://github.com/CarperAI/trlx
description = A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)
long_description = file: README.md
long_description_content_type = text/markdown
license = MIT
[options]
packages = find:
install_requires =
accelerate>=0.17.1
attrs>=22.1.0
cattrs>=22.2.0
datasets
deepspeed>=0.8.1
einops>=0.4.1
numpy>=1.23.2
torchtyping
transformers>=4.21.2
tqdm
rich
wandb>=0.13.5
tabulate>=0.9.0
networkx
tritonclient
ray@https://ray-ci-artifact-branch-public.s3.amazonaws.com/42bb0357a6fb13e4994789c824f3623f32869ad8/tmp/artifacts/.whl/ray-3.0.0.dev0-cp38-cp38-manylinux2014_x86_64.whl ; python_version=="3.8"
ray@https://ray-ci-artifact-branch-public.s3.amazonaws.com/42bb0357a6fb13e4994789c824f3623f32869ad8/tmp/artifacts/.whl/ray-3.0.0.dev0-cp39-cp39-manylinux2014_x86_64.whl ; python_version=="3.9"
ray@https://ray-ci-artifact-branch-public.s3.amazonaws.com/42bb0357a6fb13e4994789c824f3623f32869ad8/tmp/artifacts/.whl/ray-3.0.0.dev0-cp310-cp310-manylinux2014_x86_64.whl ; python_version=="3.10"
[options.extras_require]
bnb = bitsandbytes
dev =
black
hypothesis
isort
flake8
pre-commit
pytest
pytest-cov
[options.packages.find]
exclude =
docs*
tests*
[flake8]
max-complexity = 10
max-line-length = 127
# flake8 error codes: https://flake8.pycqa.org/en/latest/user/error-codes.html
# pycodestyle codes: https://pycodestyle.pycqa.org/en/latest/intro.html#error-codes
# E203 # whitespace before ‘,’, ‘;’, or ‘:’
# E741 # do not use variables named ‘l’, ‘O’, or ‘I’
# F401 # module imported but unused
# F821 # undefined name name
# W503 # line break before binary operator
# W605 # invalid escape sequence ‘x’
ignore =
E203
E741
F821
W503
W605
per-file-ignores = __init__.py:F401,loading.py:F401
exclude =
.git
__pycache__
docs/source/conf.py
build
dist