maxtext.integration.tunix.weight_mapping.deepseek3 module

maxtext.integration.tunix.weight_mapping.deepseek3 module#

Mapping MaxText Deepseek (MoE) weights to vLLM/tpu-inference keys.

class maxtext.integration.tunix.weight_mapping.deepseek3.DEEPSEEK_VLLM_MAPPING[source]#

Bases: object

Mapping MaxText Deepseek-V3 weights to Tunix/vLLM NNX keys.

static to_hf_hook_fns()[source]#
static to_hf_transpose_keys()[source]#

Returns a list of keys for weights that need to be transposed.

Returns:

An empty dictionary, as no keys require transposition for this mapping.

static lora_to_hf_mappings()[source]#

Provides the mapping for LoRA (Low-Rank Adaptation) weights.

Returns:

None, as LoRA mappings are not defined for this model.

static to_hf_mapping()[source]#

Returns the weight mapping for the model.