mirror of
https://github.com/huggingface/trl.git
synced 2025-10-20 18:43:52 +08:00
Exclude vllm dependencies from dev extra (#4229)
This commit is contained in:
committed by
GitHub
parent
d1d0407d3c
commit
e2c97a805a
@ -90,26 +90,32 @@ vlm = [
|
||||
"num2words==0.5.14"
|
||||
]
|
||||
dev = [
|
||||
# bco
|
||||
"scikit-learn",
|
||||
"joblib",
|
||||
# deepspeed
|
||||
"deepspeed>=0.14.4",
|
||||
# judges
|
||||
"openai>=1.23.2",
|
||||
"llm-blender>=0.0.2",
|
||||
# liger
|
||||
"liger-kernel>=0.6.2",
|
||||
# peft
|
||||
"peft>=0.8.0",
|
||||
# quality
|
||||
"pre-commit",
|
||||
"hf-doc-builder",
|
||||
# quantization
|
||||
"bitsandbytes",
|
||||
# scikit: included in bco
|
||||
# test
|
||||
"parameterized",
|
||||
"pytest-cov",
|
||||
"pytest-rerunfailures==15.1",
|
||||
"pytest-xdist",
|
||||
"pytest",
|
||||
"vllm==0.10.2",
|
||||
"fastapi",
|
||||
"pydantic",
|
||||
"requests",
|
||||
"uvicorn",
|
||||
# vllm: not included in dev by default due to CUDA error; see GH-4228
|
||||
# vlm
|
||||
"Pillow",
|
||||
"torchvision",
|
||||
"num2words==0.5.14"
|
||||
|
Reference in New Issue
Block a user