Unverified Commit f9076a5a authored by Zhiqiang Xie's avatar Zhiqiang Xie Committed by GitHub
Browse files

hot fix for mooncake batch set api (#9836)

parent 646076b7
......@@ -159,6 +159,7 @@ class MooncakeStore(HiCacheStorage):
def batch_set(
self,
keys: List[str],
values: Optional[List[torch.Tensor]] = None,
target_location: Optional[List[int]] = None,
target_sizes: Optional[List[int]] = None,
) -> bool:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment