diff --git a/pyproject.toml b/pyproject.toml index 806a10baac..e9e8767c8f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -223,6 +223,7 @@ include = [ "flashinfer/jit/cpp_ext_hip.py", "flashinfer/jit/env.py", "flashinfer/jit/utils.py", + "flashinfer/logits_processor/__init__.py", "flashinfer/norm.py", "flashinfer/page.py", "flashinfer/prefill_rocm.py",