Fix typos

Signed-off-by: yuantao <2422264527@qq.com>
This commit is contained in:
yuantao 2025-12-13 16:25:38 +08:00
parent b565203d92
commit 2470548aaa
2 changed files with 2 additions and 2 deletions

View File

@ -787,7 +787,7 @@ class CrossAttentionManager(SingleTypeKVCacheManager):
class SinkFullAttentionManager(FullAttentionManager):
def __init__(
self,
kv_cache_spec: KVCacheSpec,
kv_cache_spec: SinkFullAttentionSpec,
block_pool: BlockPool,
kv_cache_group_id: int,
dcp_world_size: int = 1,

View File

@ -299,7 +299,7 @@ class CrossAttentionSpec(AttentionSpec):
return cdiv(max_encoder_len, self.block_size) * self.page_size_bytes
@dataclass(forzen=True)
@dataclass(frozen=True)
class SinkFullAttentionSpec(FullAttentionSpec):
sink_len: int | None = None