o
    Ti                     @   s.   d dl mZ d dlmZ G dd dejZdS )    )partition_parameters) AsyncPartitionedParameterSwapperc                       s(   e Zd Zddeddf fddZ  ZS )QuantizationContextNparam_swapperreturnc                    s   t  j||d d S )N)config_dict_or_pathr   )super__init__)selfr   r   	__class__ i/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/inference/quantization/quantization_context.pyr	      s   zQuantizationContext.__init__)N)__name__
__module____qualname__r   r	   __classcell__r   r   r   r   r   
   s     r   N)deepspeed.runtime.zeror   7deepspeed.runtime.swap_tensor.partitioned_param_swapperr   Initr   r   r   r   r   <module>   s   