diff --git a/python/sglang/check_env.py b/python/sglang/check_env.py index d525da9292f7..1870e3207ae7 100644 --- a/python/sglang/check_env.py +++ b/python/sglang/check_env.py @@ -20,7 +20,7 @@ def is_cuda_v2(): PACKAGE_LIST = [ "sglang", "sgl_kernel", - "flashinfer", + "flashinfer_python", "triton", "transformers", "torchao", @@ -36,8 +36,8 @@ def is_cuda_v2(): "packaging", "psutil", "pydantic", - "multipart", - "zmq", + "python-multipart", + "pyzmq", "torchao", "uvicorn", "uvloop",