diff --git a/third_party/vllm/vllm/distributed/kv_transfer/kv_connector/v1/mooncake/mooncake_utils.py b/third_party/vllm/vllm/distributed/kv_transfer/kv_connector/v1/mooncake/mooncake_utils.py index e1d9862..b295121 100644 --- a/third_party/vllm/vllm/distributed/kv_transfer/kv_connector/v1/mooncake/mooncake_utils.py +++ b/third_party/vllm/vllm/distributed/kv_transfer/kv_connector/v1/mooncake/mooncake_utils.py @@ -232,7 +232,7 @@ class MooncakeBootstrapServer: after init_none_hash is called). """ import vllm.v1.core.kv_cache_utils as kv_utils - from vllm.utils.hashing import sha256_cbor + from vllm.utils.hashing import sha256 block_size = self._kv_info.get("block_size", 512) if self._kv_info else 512 n = num_tokens or len(token_ids) @@ -246,7 +246,7 @@ class MooncakeBootstrapServer: for i in range(num_blocks): block_tokens = tuple(token_ids[i * block_size:(i + 1) * block_size]) block_hash = kv_utils.hash_block_tokens( - sha256_cbor, prev_hash, block_tokens, None) + sha256, prev_hash, block_tokens, None) prev_hash = block_hash bid = self._hash_table.get(block_hash.hex())