1234567 |
- '''Copyright The Microsoft DeepSpeed Team'''
- from .elasticity import compute_elastic_config, elasticity_enabled, ensure_immutable_elastic_config
- from .utils import is_torch_elastic_compatible
- from .constants import ENABLED, ENABLED_DEFAULT, ELASTICITY
- if is_torch_elastic_compatible():
- from .elastic_agent import DSElasticAgent
|