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

Sft flash mask #8664

Merged
merged 11 commits into from
Jun 28, 2024
Merged

Sft flash mask #8664

merged 11 commits into from
Jun 28, 2024

Conversation

wtmlon
Copy link
Collaborator

@wtmlon wtmlon commented Jun 26, 2024

PR types

PR changes

Description

适配 flash mask 数据流,支持 SFT,DPO
image

Copy link

paddle-bot bot commented Jun 26, 2024

Thanks for your contribution!

Copy link

codecov bot commented Jun 26, 2024

Codecov Report

Attention: Patch coverage is 2.94118% with 33 lines in your changes missing coverage. Please review.

Project coverage is 55.61%. Comparing base (3ebe938) to head (1f6bfd1).

Files Patch % Lines
paddlenlp/data/data_collator.py 0.00% 11 Missing ⚠️
paddlenlp/trl/trl_data.py 0.00% 10 Missing ⚠️
paddlenlp/datasets/zero_padding_dataset.py 0.00% 4 Missing ⚠️
paddlenlp/trl/dpo_trainer.py 0.00% 4 Missing ⚠️
paddlenlp/transformers/llama/fusion_ops.py 0.00% 2 Missing ⚠️
paddlenlp/transformers/llama/modeling.py 33.33% 2 Missing ⚠️
Additional details and impacted files
@@             Coverage Diff             @@
##           develop    #8664      +/-   ##
===========================================
- Coverage    55.62%   55.61%   -0.01%     
===========================================
  Files          620      620              
  Lines        96949    96965      +16     
===========================================
+ Hits         53929    53930       +1     
- Misses       43020    43035      +15     

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

llm/run_finetune.py Outdated Show resolved Hide resolved
llm/run_finetune.py Outdated Show resolved Hide resolved
paddlenlp/data/data_collator.py Show resolved Hide resolved
paddlenlp/transformers/llama/modeling.py Outdated Show resolved Hide resolved
lugimzzz
lugimzzz previously approved these changes Jun 27, 2024
Copy link
Contributor

@lugimzzz lugimzzz left a comment

Choose a reason for hiding this comment

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

lgtm

@ZeyuChen ZeyuChen merged commit c86e447 into PaddlePaddle:develop Jun 28, 2024
6 of 11 checks passed
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.

5 participants