зеркало из https://github.com/microsoft/DeepSpeed.git
11 строки
444 B
Python
Executable File
11 строки
444 B
Python
Executable File
# Copyright (c) Microsoft Corporation.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
# DeepSpeed Team
|
|
|
|
from .replace_module import replace_transformer_layer, revert_transformer_layer, ReplaceWithTensorSlicing, GroupQuantizer, generic_injection
|
|
from .module_quantize import quantize_transformer_layer
|
|
from .replace_policy import HFBertLayerPolicy
|
|
from .layers import LinearAllreduce, LinearLayer, EmbeddingLayer, Normalize
|
|
from .policy import DSPolicy
|