Spaces:
Runtime error
Runtime error
[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 | |