o TÃiã@s>ddlmZddlmZmZddlmZ Gdd„deƒZdS)é)ÚOptionalé)Ú DtypeEnumÚ NormTypeEnum)ÚDSModuleConfigc@sfeZdZUdZejZeed< eed< dZ e ed< dZ eed< dZ e ed< d Z eeed <d S) ÚDSEmbeddingsConfigz( Config class for DSEmbeddings. Úresidual_dtypeÚ embedding_dimFÚpositional_embeddingrÚpositional_offsetÚuse_token_typeNÚoutput_normalization)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚfp16rÚ__annotations__Úintr Úboolr r r rr©rrúk/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/inference/v2/modules/configs/embedding_config.pyr"s    rN)ÚtypingrÚinference_utilsrrÚmodules.ds_modulerrrrrrÚs