[VERL Use Case] Add torch_memory_saver into deps (#6247)

This commit is contained in:
Stefan He
2025-05-12 19:09:03 -07:00
committed by GitHub
parent 3c32895cbe
commit 1ab14c4c5c
2 changed files with 1 additions and 2 deletions

View File

@@ -103,7 +103,7 @@ test = [
"accelerate",
"peft",
]
all = ["sglang[srt]", "sglang[openai]", "sglang[anthropic]", "sglang[litellm]"]
all = ["sglang[srt]", "sglang[openai]", "sglang[anthropic]", "sglang[litellm]", "sglang[torch_memory_saver]"]
all_hip = ["sglang[srt_hip]", "sglang[openai]", "sglang[anthropic]", "sglang[litellm]"]
all_xpu = ["sglang[srt_xpu]", "sglang[openai]", "sglang[anthropic]", "sglang[litellm]"]
all_hpu = ["sglang[srt_hpu]", "sglang[openai]", "sglang[anthropic]", "sglang[litellm]"]