config: six more runtime readers ask the bags (#36973)
Co-authored-by: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 5
parent
b65e677e48
commit
1a3e152f03
@@ -504,7 +504,6 @@ class ModelRunner:
|
||||
)
|
||||
set_global_expert_distribution_recorder(
|
||||
ExpertDistributionRecorder.init_new(
|
||||
self.server_args,
|
||||
get_global_expert_location_metadata(),
|
||||
rank=global_ep_rank,
|
||||
)
|
||||
@@ -709,7 +708,6 @@ class ModelRunner:
|
||||
)
|
||||
set_global_expert_distribution_recorder(
|
||||
ExpertDistributionRecorder.init_new(
|
||||
self.server_args,
|
||||
get_global_expert_location_metadata(),
|
||||
rank=expert_rank,
|
||||
)
|
||||
@@ -750,7 +748,6 @@ class ModelRunner:
|
||||
def maybe_init_expert_backup_client(self):
|
||||
self.expert_backup_client = (
|
||||
ExpertBackupClient(
|
||||
server_args=self.server_args,
|
||||
model_config=self.model_config,
|
||||
moe_ep_size=self.ps.moe_ep_size,
|
||||
moe_ep_rank=self.ps.moe_ep_rank,
|
||||
@@ -971,7 +968,6 @@ class ModelRunner:
|
||||
)
|
||||
set_global_expert_distribution_recorder(
|
||||
ExpertDistributionRecorder.init_new(
|
||||
self.server_args,
|
||||
get_global_expert_location_metadata(),
|
||||
rank=global_ep_rank,
|
||||
)
|
||||
@@ -1967,7 +1963,6 @@ class ModelRunner:
|
||||
return
|
||||
set_global_expert_distribution_recorder(
|
||||
ExpertDistributionRecorder.init_new(
|
||||
self.server_args,
|
||||
get_global_expert_location_metadata(),
|
||||
rank=self._elastic_global_rank(),
|
||||
)
|
||||
@@ -2032,7 +2027,6 @@ class ModelRunner:
|
||||
ElasticEPStateManager.on_scale(effective_size, target_size)
|
||||
set_global_expert_distribution_recorder(
|
||||
ExpertDistributionRecorder.init_new(
|
||||
self.server_args,
|
||||
get_global_expert_location_metadata(),
|
||||
rank=self._elastic_global_rank(),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user