o
    TÃi  ã                   @   s0   d dl mZmZ d dlmZ G dd„ deƒZdS )é   )ÚActivationTypeÚ	DtypeEnum)ÚDSModuleConfigc                   @   sT   e Zd ZU dZeed< 	 eed< 	 ejZeed< 	 e	j
Ze	ed< 	 e	j
Ze	ed< dS )ÚDSLinearConfigz(
    Config class for DSLinearBase.
    Úin_channelsÚout_channelsÚ
activationÚinput_dtypeÚoutput_dtypeN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚintÚ__annotations__r   ÚIDENTITYr   r   Úfp16r	   r
   © r   r   úh/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/inference/v2/modules/configs/linear_config.pyr   
   s   
 r   N)Úinference_utilsr   r   Úmodules.ds_moduler   r   r   r   r   r   Ú<module>   s   