File size: 611 Bytes
b8c8910 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | # Copyright (c) Microsoft Corporation.
# SPDX-License-Identifier: Apache-2.0
# DeepSpeed Team
from .partition_parameters import ZeroParamType
from .partition_parameters import ZeroParamStatus
from .partition_parameters import Init
from .partition_parameters import GatheredParameters
from .partition_parameters import register_external_parameter
from .parameter_offload import DeepSpeedZeRoOffload
from .partition_parameters import DeepSpeedTensorOverride
from .tiling import TiledLinear
from .tiling import TiledLinearReturnBias
from .mics import MiCS_Init
from .stage3 import unwrap_model_for_generation
|