mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-07 09:31:35 +08:00
support c4 attn && fix cache
This commit is contained in:
@@ -132,6 +132,7 @@ class IluvatarAttnBackend(AttentionBackend):
|
||||
def get_kv_cache_shape(
|
||||
self,
|
||||
max_num_blocks: int,
|
||||
kv_cache_quant_type: str = None,
|
||||
):
|
||||
"""
|
||||
Caculate kv cache shape
|
||||
|
Reference in New Issue
Block a user