wehub-resource-sync
20 lines
611 B
Python
20 lines
611 B
Python
# 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
|