chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,19 @@
|
||||
# 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
|
||||
Reference in New Issue
Block a user