update bid to match each layers MTP source (#23237)

* update bid to match each layers MTP source

* Update conversion/qwen.py

Co-authored-by: Sigbjørn Skjæret <sigbjorn.skjaeret@scala.com>

---------

Co-authored-by: Sigbjørn Skjæret <sigbjorn.skjaeret@scala.com>
This commit is contained in:
Incarnas
2026-05-17 21:37:12 -07:00
committed by GitHub
parent dd7cad7197
commit 1867a0c692
+1
View File
@@ -600,6 +600,7 @@ class _Qwen35MtpMixin:
if name.find("layers.") != -1: if name.find("layers.") != -1:
assert bid is not None assert bid is not None
name = name.replace(f"mtp.layers.{bid}", f"model.layers.{bid + n_layer}") name = name.replace(f"mtp.layers.{bid}", f"model.layers.{bid + n_layer}")
bid = bid + n_layer
else: else:
remapper = { remapper = {
"mtp.fc": "model.layers.{bid}.eh_proj", "mtp.fc": "model.layers.{bid}.eh_proj",