fix slow load

[lugimzzz]

PR types

Bug fixes

PR changes

APIs

Description

对大模型调用state_dict()耗时长，避免多分片加载时候反复调用state_dict()

[paddle-bot]

Thanks for your contribution!

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 48.67%. Comparing base (53273ad) to head (5b46064).
Report is 101 commits behind head on develop.

❌ Your project status has failed because the head coverage (48.67%) is below the target coverage (58.00%). You can increase the head coverage or adjust the target coverage.

Additional details and impacted files

@@             Coverage Diff             @@
##           develop   #10498      +/-   ##
===========================================
- Coverage    48.92%   48.67%   -0.26%     
===========================================
  Files          767      768       +1     
  Lines       126238   126916     +678     
===========================================
+ Hits         61766    61780      +14     
- Misses       64472    65136     +664     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.