Skip to content

Commit 4c8fd1a

Browse files
author
Weichao Luo
committed
fix name.
1 parent 5bd5ffd commit 4c8fd1a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

lightllm/server/router/model_infer/mode_backend/continues_batch/pd_mode/prefill_node_impl/prefill_kv_move_manager.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -329,9 +329,9 @@ def __init__(self, args, info_queue: mp.Queue, mem_queues: List[mp.Queue]):
329329
# start a single kv trans process
330330
self.kv_trans_task_in_queue = mp.Queue()
331331
self.kv_trans_task_out_queue = mp.Queue()
332-
from .prefill_trans_process import start_decode_trans_process
332+
from .prefill_trans_process import start_prefill_trans_process
333333
self.kv_trans_port = find_available_port(self.args.pd_p_allowed_port_min, self.args.pd_p_allowed_port_max)
334-
self.kv_trans_process = start_decode_trans_process(
334+
self.kv_trans_process = start_prefill_trans_process(
335335
self.args, self.host_ip, self.kv_trans_port, self.kv_trans_task_in_queue, self.kv_trans_task_out_queue, self.mem_queues)
336336

337337
assert self.kv_trans_task_out_queue.get(timeout=30) == "proc_start"

lightllm/server/router/model_infer/mode_backend/continues_batch/pd_mode/prefill_node_impl/prefill_trans_process.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def _init_env(
8787
pass
8888

8989

90-
def start_decode_trans_process(
90+
def start_prefill_trans_process(
9191
args,
9292
store_ip,
9393
store_port,

0 commit comments

Comments
 (0)