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

[Loader]Fix bug in MTP weight loading (#5744)

Browse Source 
* fix torch mtp

* fix

* update
This commit is contained in:
bukejiyu
2025-12-25 11:32:17 +08:00
committed by GitHub
parent e154c03416
commit f0bbdce849
6 changed files with 26 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/model_executor/model_loader/default_loader_v1.py
+2 -2
View File
@@ -56,8 +56,8 @@ class DefaultModelLoaderV1(BaseModelLoader):
load_weights_from_cache(model, weights_iterator)
else:
model.load_weights(weights_iterator)
if fd_config.speculative_config.model_type != "mtp":
process_final_after_loading(model, fd_config)
process_final_after_loading(model, fd_config)
self.clean_memory_fragments()