Spaces:
Runtime error
Runtime error
File size: 1,572 Bytes
fa6856c |
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 |
[metadata]
name = trlx
author = Alex Havrilla
version = 0.7.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.27.1
tqdm
rich
wandb>=0.13.5
ray>=2.4.0
tabulate>=0.9.0
networkx
tritonclient
[options.extras_require]
bnb =
bitsandbytes
scipy # fix(bnb): Remove when `bitsandbytes` adds this dependency
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
|