[Unified Checkpoint] Fix tie_weights save and load

[DesmonDay]

PR types

Bug fixes

PR changes

Others

Description

Fix tie_weights save and load.

[paddle-bot]

Thanks for your contribution!

[codecov]

Codecov Report

Attention: Patch coverage is 7.69231% with 12 lines in your changes are missing coverage. Please review.

Project coverage is 55.37%. Comparing base (d83bd5e) to head (9d87445).
Report is 5 commits behind head on develop.

❗ Current head 9d87445 differs from pull request most recent head 57476c9. Consider uploading reports for the commit 57476c9 to get more accurate results

Files	Patch %	Lines
paddlenlp/trainer/plugins/unified_checkpoint.py	0.00%	11 Missing ⚠️
paddlenlp/trainer/trainer.py	0.00%	1 Missing ⚠️

Additional details and impacted files

@@             Coverage Diff             @@
##           develop    #8137      +/-   ##
===========================================
- Coverage    56.45%   55.37%   -1.08%     
===========================================
  Files          596      596              
  Lines        91602    91622      +20     
===========================================
- Hits         51711    50739     -972     
- Misses       39891    40883     +992     

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

[JunnYu]

LGTM

[ZHUI]

Suggested change

	if hasattr(model, "_tied_weights_keys") and model._tied_weights_keys is not None:
	for key in model._tied_weights_keys:
	expected_keys.remove(key)
	if model._keys_to_ignore_on_save is not None:
	for key in model._keys_to_ignore_on_save:
	expected_keys.remove(key)

[ZHUI]

PaddleNLP/paddlenlp/transformers/model_utils.py

Lines 885 to 886 in 1cd270a

	_keys_to_ignore_on_save = None
	_tied_weights_keys = None

我们的模型，应该有这个属性

[ZHUI]

用 _keys_to_ignore_on_save 可能更好一点。

[JunnYu]

LGTM

[ZHUI]

LGTM