[4/N] (Elastic EP) Back up Expert Weights in DRAM (#17374)
Co-authored-by: UNIDY2002 <unidy2002@outlook.com>
This commit is contained in:
@@ -56,7 +56,6 @@ class ElasticEPStateManager:
|
||||
def _build_state(
|
||||
cls, *, ep_size: Optional[int] = None, device: Optional[torch.device] = None
|
||||
) -> ElasticEPState:
|
||||
|
||||
active = cls.healthy_rank_state(ep_size=ep_size, device=device)
|
||||
return ElasticEPState(
|
||||
active_ranks=active,
|
||||
|
||||
Reference in New Issue
Block a user