o TÃi6ã@s<ddlmZmZddlmZddlmZGdd„deƒZdS)é)Ú DtypeEnumÚ NormTypeEnum)ÚDSModuleConfigé)ÚOptionalc@sPeZdZUdZejZeed< dZe e ed< e ed< e ed< e ed<dS)ÚDSUnembedConfigz$ Config class for DSUnembed ÚdtypeNÚ norm_typeÚ model_dimÚ max_sequencesÚ vocab_size) Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚfp16rÚ__annotations__r rrÚint©rrúi/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/inference/v2/modules/configs/unembed_config.pyr s rN)Úinference_utilsrrÚmodules.ds_modulerÚtypingrrrrrrÚs