[PD] Fix DeepSeek cannot be run on latest master (#5568)

2025-04-20 17:19:48 +08:00
parent 99456bcacb
commit 80ac527d22
1 changed files with 1 additions and 0 deletions
--- a/python/sglang/srt/mem_cache/memory_pool.py
+++ b/python/sglang/srt/mem_cache/memory_pool.py
@@ -418,6 +418,7 @@ class MLATokenToKVPool(KVCache):
        enable_memory_saver: bool,
    ):
        self.size = size
+        self.page_size = page_size
        self.dtype = dtype
        self.device = device
        if dtype in (torch.float8_e5m2, torch.float8_e4m3fn):