o
    TÃiÍ  ã                   @   s`   d dl mZmZmZmZmZ d dlmZ d dlm	Z	 d dl
mZmZmZmZmZ d dlmZ dS )é   )Úreplace_transformer_layerÚrevert_transformer_layerÚReplaceWithTensorSlicingÚGroupQuantizerÚgeneric_injection)Úquantize_transformer_layer)ÚHFBertLayerPolicy)ÚLinearAllreduceÚLinearLayerÚEmbeddingLayerÚ	NormalizeÚset_autotp_mode)ÚDSPolicyN)Úreplace_moduler   r   r   r   r   Úmodule_quantizer   Úreplace_policyr   Úlayersr	   r
   r   r   r   Úpolicyr   © r   r   úT/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/module_inject/__init__.pyÚ<module>   s
   