Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

llama beam search fix #6882

Merged
merged 5 commits into from
Sep 1, 2023
Merged

Conversation

wtmlon
Copy link
Collaborator

@wtmlon wtmlon commented Aug 31, 2023

PR types

PR changes

Description

修正llama beam search 错误

@paddle-bot
Copy link

paddle-bot bot commented Aug 31, 2023

Thanks for your contribution!

sijunhe
sijunhe previously approved these changes Aug 31, 2023
@codecov
Copy link

codecov bot commented Aug 31, 2023

Codecov Report

Merging #6882 (03db007) into develop (3b41581) will decrease coverage by 0.01%.
Report is 4 commits behind head on develop.
The diff coverage is 100.00%.

@@             Coverage Diff             @@
##           develop    #6882      +/-   ##
===========================================
- Coverage    59.93%   59.93%   -0.01%     
===========================================
  Files          547      547              
  Lines        80996    81003       +7     
===========================================
- Hits         48548    48547       -1     
- Misses       32448    32456       +8     
Files Changed Coverage Δ
paddlenlp/transformers/generation_utils.py 69.06% <100.00%> (+0.07%) ⬆️

... and 6 files with indirect coverage changes

@@ -220,6 +221,12 @@ def test_model_attention_mask(self):
config_and_inputs = self.model_tester.prepare_config_and_inputs()
self.model_tester.create_and_check_model_attention_mask(*config_and_inputs)

def test_beam_search_generate(self):
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

你这里写一下 TODO 吧,说明一下当前问题,后续需要修复。

@sijunhe sijunhe merged commit 78583b2 into PaddlePaddle:develop Sep 1, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants