click==8.1.7
torch==2.1.2
scipy==1.10.1
wandb==0.16.3
protobuf==3.20.2
urllib3<2,>=1.26.18
pydantic==2.6.0
pydantic-yaml==1.2.0
ray[default]==2.9.3
loguru==0.7.2
s3fs
datasets>=2.17.1
transformers==4.36.2
accelerate==0.26.1
peft==0.7.1
trl==0.7.10
bitsandbytes==0.42.0
bert_score==0.3.13
lm-eval==0.4.2
einops==0.7.0
fschat==0.2.36
openai==1.14.3
ragas==0.1.5
langchain-community==0.0.29
langchain_openai==0.1.1
sentencepiece==0.2.0

[dev]
lm-buddy[docs,ruff,test]
pre-commit==3.6.0
jupyter==1.0.0

[docs]
Sphinx==7.2.6
nbsphinx==0.9.3
sphinx-codeautolink==0.15.0
myst-parser==2.0.0
furo==2024.1.29

[ruff]
ruff==0.2.1

[test]
pytest==7.4.3
pytest-cov==4.1.0
