hot fix for mooncake batch set api (#9836)
This commit is contained in:
@@ -159,6 +159,7 @@ class MooncakeStore(HiCacheStorage):
|
|||||||
def batch_set(
|
def batch_set(
|
||||||
self,
|
self,
|
||||||
keys: List[str],
|
keys: List[str],
|
||||||
|
values: Optional[List[torch.Tensor]] = None,
|
||||||
target_location: Optional[List[int]] = None,
|
target_location: Optional[List[int]] = None,
|
||||||
target_sizes: Optional[List[int]] = None,
|
target_sizes: Optional[List[int]] = None,
|
||||||
) -> bool:
|
) -> bool:
|
||||||
|
|||||||
Reference in New Issue
Block a user