[Feature] LMCache Connector Integration (#9741)
Signed-off-by: Oasis-Git <ayw.sirius19@gmail.com> Signed-off-by: YuhanLiu11 <yliu738@wisc.edu> Co-authored-by: Zhiqiang Xie <xiezhq@stanford.edu>
This commit is contained in:
@@ -369,7 +369,6 @@ class MHATokenToKVPool(KVCache):
|
||||
# same applies to get_value_buffer and get_kv_buffer
|
||||
if self.layer_transfer_counter is not None:
|
||||
self.layer_transfer_counter.wait_until(layer_id - self.start_layer)
|
||||
|
||||
return self._get_key_buffer(layer_id)
|
||||
|
||||
def _get_value_buffer(self, layer_id: int):
|
||||
|
||||
Reference in New Issue
Block a user