diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py index 7501f6e0c..31294749d 100644 --- a/python/sglang/srt/managers/scheduler.py +++ b/python/sglang/srt/managers/scheduler.py @@ -259,7 +259,7 @@ class Scheduler( self.enable_metrics_for_all_schedulers = ( server_args.enable_metrics_for_all_schedulers ) - self.enable_kv_cache_events = server_args.kv_events_config is not None + self.enable_kv_cache_events = server_args.kv_events_config and tp_rank == 0 self.stream_interval = server_args.stream_interval self.spec_algorithm = SpeculativeAlgorithm.from_string( server_args.speculative_algorithm @@ -553,9 +553,11 @@ class Scheduler( # Init metrics stats self.init_metrics(tp_rank, pp_rank, dp_rank) - self.init_kv_events(server_args.kv_events_config) self.init_dp_balance(dp_balance_meta) + if self.enable_kv_cache_events: + self.init_kv_events(server_args.kv_events_config) + # Init disaggregation self.disaggregation_mode = DisaggregationMode( self.server_args.disaggregation_mode