mimo2: address MTP review feedback (#26228)

Co-authored-by: tnhnyc <115956684+tnhnyc@users.noreply.github.com>
This commit is contained in:
Tunahan
2026-07-30 11:55:58 +08:00
committed by GitHub
co-authored by tnhnyc
parent 3018a11e79
commit 64d528be72
4 changed files with 191 additions and 22 deletions
+3
View File
@@ -4440,8 +4440,11 @@ MODEL_TENSORS: dict[MODEL_ARCH, list[MODEL_TENSOR]] = {
MODEL_TENSOR.FFN_EXP_PROBS_B,
MODEL_TENSOR.LAYER_OUT_NORM,
MODEL_TENSOR.NEXTN_EH_PROJ,
MODEL_TENSOR.NEXTN_EMBED_TOKENS,
MODEL_TENSOR.NEXTN_ENORM,
MODEL_TENSOR.NEXTN_HNORM,
MODEL_TENSOR.NEXTN_SHARED_HEAD_HEAD,
MODEL_TENSOR.NEXTN_SHARED_HEAD_NORM,
],
MODEL_ARCH.STEP35: [
MODEL_TENSOR.TOKEN_EMBD,