Skip to content

Commit 6a31b88

Browse files
committed
Set scheduler v1 as default
1 parent ae74754 commit 6a31b88

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

fastdeploy/engine/args_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -396,7 +396,7 @@ def __post_init__(self):
396396
envs.ENABLE_V1_KVCACHE_SCHEDULER = 0
397397
if self.splitwise_role != "mixed":
398398
envs.ENABLE_V1_KVCACHE_SCHEDULER = 0
399-
if (not current_platform.is_cuda()) and (not current_platform.is_xpu()):
399+
if not current_platform.is_cuda():
400400
envs.ENABLE_V1_KVCACHE_SCHEDULER = 0
401401

402402
@staticmethod

fastdeploy/worker/worker_process.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -758,7 +758,7 @@ def initialize_fd_config(args, ranks: int = 1, local_rank: int = 0) -> FDConfig:
758758
if args.splitwise_role != "mixed":
759759
logger.info(f"Set ENABLE_V1_KVCACHE_SCHEDULER to 0 due to not supported {args.splitwise_role} now.")
760760
envs.ENABLE_V1_KVCACHE_SCHEDULER = 0
761-
if (not current_platform.is_cuda()) and (not current_platform.is_xpu()):
761+
if not current_platform.is_cuda():
762762
logger.info("Set ENABLE_V1_KVCACHE_SCHEDULER to 0 due to not supported.")
763763
envs.ENABLE_V1_KVCACHE_SCHEDULER = 0
764764

0 commit comments

Comments
 (0)