[console_scripts]
retrieval-heads.detect = retrieval_heads.cli:detect_cli
retrieval-heads.niah = retrieval_heads.cli:niah_cli
retrieval-heads.visualize = retrieval_heads.visualize:cli

[vllm.general_plugins]
gdn_qktap_plugin = retrieval_heads:register
