add missing dep

This commit is contained in:
Dan Saunders
2025-10-03 12:46:15 -04:00
parent 26418e6f9a
commit 786f1a3ff9
2 changed files with 234 additions and 140 deletions

View File

@@ -107,6 +107,7 @@ optimizers = [
ray = ["ray[train]"]
vllm = ["vllm>=0.10.0"]
llmcompressor = ["llmcompressor>=0.5.1"]
fbgemm-gpu = ["fbgemm-gpu-genai>=1.2.0"]
dev = [
"pytest",
"pytest-cov",