wehub-resource-sync
14 lines
514 B
Python
14 lines
514 B
Python
# Copyright (c) Microsoft Corporation.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
# DeepSpeed Team
|
|
|
|
from deepspeed.runtime.zero import partition_parameters
|
|
from deepspeed.runtime.swap_tensor.partitioned_param_swapper import AsyncPartitionedParameterSwapper
|
|
|
|
|
|
class QuantizationContext(partition_parameters.Init):
|
|
|
|
def __init__(self, config_dict_or_path, param_swapper: AsyncPartitionedParameterSwapper = None) -> None:
|
|
super().__init__(config_dict_or_path=config_dict_or_path, param_swapper=param_swapper)
|