Skip to content

GitLab

Unverified Commit bc1f6fda authored by Yineng Zhang's avatar Yineng Zhang Committed by GitHub
Browse files

fix: add cuda-python for xgrammar (#2199)

parent 867e092f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/pyproject.toml
View file @ bc1f6fda
...@@ -23,7 +23,7 @@ runtime_common = ["aiohttp", "decord", "fastapi", ...@@ -23,7 +23,7 @@ runtime_common = ["aiohttp", "decord", "fastapi",
"psutil", "pydantic", "python-multipart", "psutil", "pydantic", "python-multipart",
"pyzmq>=25.1.2", "torchao", "uvicorn", "uvloop", "pyzmq>=25.1.2", "torchao", "uvicorn", "uvloop",
"xgrammar>=0.1.4"] "xgrammar>=0.1.4"]
srt = ["sglang[runtime_common]", "torch", "vllm>=0.6.3.post1"] srt = ["sglang[runtime_common]", "torch", "vllm>=0.6.3.post1", "cuda-python"]
# HIP (Heterogeneous-computing Interface for Portability) for AMD # HIP (Heterogeneous-computing Interface for Portability) for AMD
# => base docker rocm/vllm-dev:20241022, not from public vllm whl # => base docker rocm/vllm-dev:20241022, not from public vllm whl
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment