apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-01 06:42:23 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix bug for scheduler v0 (#3307)

Browse Source
This commit is contained in:
chenjian
2025-08-11 23:55:20 +08:00
committed by GitHub
parent a84a98b107
commit 4646aff25c
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fastdeploy/output/token_processor.py
View File

@@ -463,6 +463,7 @@ class TokenProcessor:
if recovery_stop: if recovery_stop:
llm_logger.info(f"recovery stop signal found at task {task_id}") llm_logger.info(f"recovery stop signal found at task {task_id}")
if not recovery_stop and token_id < 0: if not recovery_stop and token_id < 0:
if envs.ENABLE_V1_KVCACHE_SCHEDULER:
if task_id in self.resource_manager.to_be_rescheduled_request_id_set: if task_id in self.resource_manager.to_be_rescheduled_request_id_set:
self.resource_manager.reschedule_preempt_task(task_id) self.resource_manager.reschedule_preempt_task(task_id)
continue continue