-
Notifications
You must be signed in to change notification settings - Fork 3k
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
add llama & qwen dpo #8474
add llama & qwen dpo #8474
Conversation
Thanks for your contribution! |
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## develop #8474 +/- ##
===========================================
- Coverage 54.29% 54.02% -0.27%
===========================================
Files 617 621 +4
Lines 96339 96878 +539
===========================================
+ Hits 52310 52343 +33
- Misses 44029 44535 +506 ☔ View full report in Codecov by Sentry. |
llm/utils.py
Outdated
@@ -211,13 +211,6 @@ def prediction_step( | |||
# keepdim in order to maintain the same shape as logits | |||
if isinstance(logits, (list, tuple)): | |||
logits = logits[0] | |||
# all gather logits when enabling tensor_parallel_output |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里删除tensor parallel output的支持的原因是什么?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
删错了,需要恢复
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
PR types
New features
PR changes
APIs
Description
新增dpo