apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 00:17:25 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

[Cherry-Pick][Loader]Fix bug in MTP weight loading #5744 (#5746)

Browse Source 
* fix mtp

* fix

---------

Co-authored-by: Jiang-Jia-Jun <163579578+Jiang-Jia-Jun@users.noreply.github.com>
This commit is contained in:
bukejiyu
2025-12-25 14:21:23 +08:00
committed by GitHub
parent 47ffaa41b1
commit e627e13808
4 changed files with 23 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/model_executor/utils.py
+7
View File
@@ -166,6 +166,13 @@ class WeightsMapper:
return self._map_name(weight_name)
def remap_weight_keys(weights_iterator, mapper: dict):
return (
(next((key.replace(k, v) for k, v in mapper.items() if k in key), key), value)
for key, value in weights_iterator
)
def process_weights_before_loading(
*, skip_prefixes: Optional[List[str]] = None, mapper: Optional[WeightsMapper] = None
):