Skip to content

Commit

Permalink
fix (#9612)
Browse files Browse the repository at this point in the history
  • Loading branch information
DrownFish19 authored Dec 11, 2024
1 parent b205922 commit 272e715
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions paddlenlp/transformers/model_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -864,6 +864,7 @@ def _load_state_dict_into_model(model_to_load, state_dict, start_prefix):
warnings.resetwarnings()
# paddlenlp hold missing_keys , just ignore not found warnings.
warnings.filterwarnings("ignore", message=r".*is not found in the provided dict.*")
warnings.filterwarnings("ignore", message=r".*paddle.to_tensor.*")
model_to_load.set_state_dict(state_dict)
error_msgs.extend([str(x.message) for x in w])

Expand Down

0 comments on commit 272e715

Please sign in to comment.