diff --git a/python/pyproject.toml b/python/pyproject.toml index 397ca3a95..d916fcb57 100644 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -72,6 +72,7 @@ blackwell = [ "cuda-python", "einops", "flashinfer_python==0.2.9rc2", + "tiktoken", ] # HIP (Heterogeneous-computing Interface for Portability) for AMD