Fix torch.cuda.MemPool() internal assertion failure (#4687)

Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>

Fix torch.cuda.MemPool() internal assertion failure (#4687)
Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
d3f71f5e · fzyzcjy · GitHub · 5eae67cb · d3f71f5e
Unverified Commit d3f71f5e authored Mar 28, 2025 by fzyzcjy Committed by GitHub Mar 27, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/pyproject.toml python/pyproject.toml +1 -1

No files found.
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -74,7 +74,7 @@ srt_cpu = ["sglang[runtime_common]", "outlines>=0.0.44,<=0.1.11", "torch"]
 openai = ["openai>=1.0", "tiktoken"]
 anthropic = ["anthropic>=0.20.0"]
 litellm = ["litellm>=1.0.0"]
-torch_memory_saver = ["torch_memory_saver>=0.0.3"]
+torch_memory_saver = ["torch_memory_saver>=0.0.4"]
 test = [
    "jsonlines",
    "matplotlib",