Skip to content

Commit ab5d933

Browse files
committed
fix too long
1 parent 4f29672 commit ab5d933

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

lightllm/common/radixmem_buffer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,9 @@ def free_req_index(self, req_id: int):
143143
def alloc(self, need_size) -> torch.Tensor:
144144
with self.lock:
145145
if need_size > self.mark_end.get_value() - self.mark_start.get_value():
146-
logger.error(f"warn no enough cache need_size {need_size} left_size {self.can_use_mem_size.get_value()}")
146+
logger.error(
147+
f"warn no enough cache need_size {need_size} left_size {self.can_use_mem_size.get_value()}"
148+
)
147149
raise RuntimeError(f"Not enough memory to allocate {need_size} tokens.")
148150

149151
start = self.mark_start.get_value()

lightllm/server/router/dynamic_prompt/hiradix/hiradix_cache.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,8 @@ def match_prefix(self, req, key, update_refs=False):
137137
max_len, query_index = self.local_cache_manager.query(req, key)
138138

139139
logger.debug(
140-
f"HiCache rank_in_node={self.rank_in_node} current key len {len(key)} match radix len {tree_node.node_prefix_total_len}, max len {max_len}"
140+
f"HiCache rank_in_node={self.rank_in_node} current key len {len(key)} match radix len "
141+
f"{tree_node.node_prefix_total_len}, max len {max_len}"
141142
)
142143
if max_len > tree_node.node_prefix_total_len:
143144
pull_len = max_len - tree_node.node_prefix_total_len
@@ -148,7 +149,8 @@ def match_prefix(self, req, key, update_refs=False):
148149
if ans_value is not None:
149150
buffers = torch.concat([ans_value, buffers])
150151
logger.debug(
151-
f"HiCache current match ratio {self.disk_cache_match_ratio.arr[0]}, pulled cache len {pull_len} from disk"
152+
f"HiCache current match ratio {self.disk_cache_match_ratio.arr[0]},"
153+
f"pulled cache len {pull_len} from disk"
152154
)
153155
res = self.local_cache_manager.read(key[:max_len], buffers, query_index, alloc_len=pull_len)
154156
if res:

0 commit comments

Comments
 (0)