frozenleaves/trl
1
0
Fork 0
mirror of https://github.com/huggingface/trl.git synced 2025-10-20 18:43:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Exclude vllm dependencies from dev extra (#4229)

Browse Source
This commit is contained in:
Albert Villanova del Moral
2025-10-08 18:14:23 +02:00
committed by GitHub
parent d1d0407d3c
commit e2c97a805a
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pyproject.toml
View File

@ -90,26 +90,32 @@ vlm = [
"num2words==0.5.14" "num2words==0.5.14"
] ]
dev = [ dev = [
# bco
"scikit-learn", "scikit-learn",
"joblib", "joblib",
# deepspeed
"deepspeed>=0.14.4", "deepspeed>=0.14.4",
# judges
"openai>=1.23.2", "openai>=1.23.2",
"llm-blender>=0.0.2", "llm-blender>=0.0.2",
# liger
"liger-kernel>=0.6.2", "liger-kernel>=0.6.2",
# peft
"peft>=0.8.0", "peft>=0.8.0",
# quality
"pre-commit", "pre-commit",
"hf-doc-builder", "hf-doc-builder",
# quantization
"bitsandbytes", "bitsandbytes",
# scikit: included in bco
# test
"parameterized", "parameterized",
"pytest-cov", "pytest-cov",
"pytest-rerunfailures==15.1", "pytest-rerunfailures==15.1",
"pytest-xdist", "pytest-xdist",
"pytest", "pytest",
"vllm==0.10.2", # vllm: not included in dev by default due to CUDA error; see GH-4228
"fastapi", # vlm
"pydantic",
"requests",
"uvicorn",
"Pillow", "Pillow",
"torchvision", "torchvision",
"num2words==0.5.14" "num2words==0.5.14"