search

vllm-project / vllm

A high-throughput and memory-efficient inference and serving engine for LLMs
https://docs.vllm.ai
Apache License 2.0
30.07k stars 4.54k forks source link

[Feature]: Add support for interchangable radix attention #6078

Open yifan1130 opened 4 months ago

yifan1130 commented 4 months ago

🚀 The feature, motivation and pitch

I am working on adjustment of radix attentions now. Thank you for your support for the radix attention. Currently, catching for A that allows for more efficient A+B generation. However, in some tree-of-thoughts settings, we are also interested for caching A+B, A+C, A+D, and thus more efficiently generates A+B+C+D. I think this feature could be developed with some adjustment from the current function of `enable_prefix_caching. I would also really appreciate if you could share some insights on how to implement this function from the current implementation of prefix_caching. Thank you very much for the great work!

Alternatives

No response

Additional context

No response

github-actions[bot] commented 2 weeks ago

This issue has been automatically marked as stale because it has not had any activity within 90 days. It will be automatically closed if no further activity occurs within 30 days. Leave a comment if you feel this issue should remain open. Thank you!