[Unified Checkpoint] Fix expert parallel

[DesmonDay]

PR types

Bug fixes

PR changes

Others

Description

Fix expert parallel model loading.

[ZHUI]

LGTM

[codecov]

Codecov Report

Attention: Patch coverage is 0% with 2 lines in your changes missing coverage. Please review.

Project coverage is 52.34%. Comparing base (3e15f0e) to head (8e85839).
Report is 267 commits behind head on develop.

Files with missing lines	Patch %	Lines
paddlenlp/trainer/unified_checkpoint/load_local.py	0.00%	2 Missing ⚠️

Additional details and impacted files

@@             Coverage Diff             @@
##           develop    #9741      +/-   ##
===========================================
- Coverage    52.45%   52.34%   -0.11%     
===========================================
  Files          721      721              
  Lines       115768   113896    -1872     
===========================================
- Hits         60728    59621    -1107     
+ Misses       55040    54275     -765     

☔ 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.