Minor improvements of TokenizerManager / health check (#6327)

2025-05-15 15:29:25 -07:00
parent cd8d4b9dfc
commit e07a6977e7
9 changed files with 136 additions and 33 deletions
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -96,12 +96,14 @@ anthropic = ["anthropic>=0.20.0"]
 litellm = ["litellm>=1.0.0"]
 torch_memory_saver = ["torch_memory_saver>=0.0.4"]
 test = [
+    "accelerate",
+    "torchaudio",
    "jsonlines",
    "matplotlib",
    "pandas",
-    "sentence_transformers",
-    "accelerate",
    "peft",
+    "timm",
+    "sentence_transformers",
 ]
 all = ["sglang[srt]", "sglang[openai]", "sglang[anthropic]", "sglang[litellm]", "sglang[torch_memory_saver]"]
 all_hip = ["sglang[srt_hip]", "sglang[openai]", "sglang[anthropic]", "sglang[litellm]"]