Fix memory access in sampling

[FrostML]

PR types

Bug fixes

PR changes

Others

Description

Fix memory access in sampling.

[paddle-bot]

Thanks for your contribution!

[codecov]

Codecov Report

Merging #4624 (77f8be0) into develop (a8a8338) will increase coverage by 0.17%.
The diff coverage is n/a.

@@             Coverage Diff             @@
##           develop    #4624      +/-   ##
===========================================
+ Coverage    41.92%   42.10%   +0.17%     
===========================================
  Files          434      433       -1     
  Lines        61824    61792      -32     
===========================================
+ Hits         25921    26017      +96     
+ Misses       35903    35775     -128     

Impacted Files	Coverage Δ
paddlenlp/transformers/nezha/modeling.py	94.36% <0.00%> (-0.64%)	⬇️
paddlenlp/utils/file_lock.py
paddlenlp/transformers/ernie/modeling.py	85.21% <0.00%> (+0.30%)	⬆️
paddlenlp/transformers/model_utils.py	77.63% <0.00%> (+1.31%)	⬆️
paddlenlp/utils/downloader.py	61.23% <0.00%> (+2.62%)	⬆️
paddlenlp/transformers/auto/modeling.py	80.32% <0.00%> (+4.01%)	⬆️
paddlenlp/ops/ext_utils.py	33.79% <0.00%> (+4.38%)	⬆️
paddlenlp/prompt/verbalizer.py	89.64% <0.00%> (+32.74%)	⬆️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.