bridge.models.deepseek.common

Module Contents

Functions

get_common_configs	Returns a dictionary of common configurations for the DeepSeek family of models.
get_common_mapping_list	Returns a list of common parameter mappings for the DeepSeek family of models.

API

bridge.models.deepseek.common.get_common_configs(

hf_pretrained: megatron.bridge.models.hf_pretrained.causal_lm.PreTrainedCausalLM,

) → dict

Returns a dictionary of common configurations for the DeepSeek family of models.

bridge.models.deepseek.common.get_common_mapping_list() → list

Returns a list of common parameter mappings for the DeepSeek family of models.