12345678910111213 |
- # 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)
|