Skip to content

Commit e2c97a8

Browse files
Exclude vllm dependencies from dev extra (#4229)
1 parent d1d0407 commit e2c97a8

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

pyproject.toml

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -90,26 +90,32 @@ vlm = [
9090
"num2words==0.5.14"
9191
]
9292
dev = [
93+
# bco
9394
"scikit-learn",
9495
"joblib",
96+
# deepspeed
9597
"deepspeed>=0.14.4",
98+
# judges
9699
"openai>=1.23.2",
97100
"llm-blender>=0.0.2",
101+
# liger
98102
"liger-kernel>=0.6.2",
103+
# peft
99104
"peft>=0.8.0",
105+
# quality
100106
"pre-commit",
101107
"hf-doc-builder",
108+
# quantization
102109
"bitsandbytes",
110+
# scikit: included in bco
111+
# test
103112
"parameterized",
104113
"pytest-cov",
105114
"pytest-rerunfailures==15.1",
106115
"pytest-xdist",
107116
"pytest",
108-
"vllm==0.10.2",
109-
"fastapi",
110-
"pydantic",
111-
"requests",
112-
"uvicorn",
117+
# vllm: not included in dev by default due to CUDA error; see GH-4228
118+
# vlm
113119
"Pillow",
114120
"torchvision",
115121
"num2words==0.5.14"

0 commit comments

Comments
 (0)