-
Notifications
You must be signed in to change notification settings - Fork 3k
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
[Unified Checkpoint] Fix generation config save #9223
[Unified Checkpoint] Fix generation config save #9223
Conversation
Thanks for your contribution! |
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## develop #9223 +/- ##
===========================================
+ Coverage 52.96% 53.09% +0.13%
===========================================
Files 657 657
Lines 106465 107030 +565
===========================================
+ Hits 56385 56831 +446
- Misses 50080 50199 +119 ☔ View full report in Codecov by Sentry. |
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
* [Unified Checkpoint] Support expert parallel (#9055) * update code * [Unified Checkpoint] Fix generation config save (#9223) * [Unified Checkpoint] update async_save_info in develop (#9173) * [Unified Checkpoint] update async save logic (#9274) * update async save signal * fix async save hang * bug fix --------- Co-authored-by: Weiguo Zhu <DrownFish19@gmail.com>
* [Unified Checkpoint] Support expert parallel (#9055) * update code * [Unified Checkpoint] Fix generation config save (#9223) * [Unified Checkpoint] update async_save_info in develop (#9173) * [Unified Checkpoint] update async save logic (#9274) * update async save signal * fix async save hang * bug fix * bug fix * [Trainer] fix save_model (#9286) * bug fix * bug fix --------- Co-authored-by: Weiguo Zhu <DrownFish19@gmail.com>
PR types
Bug fixes
PR changes
Others
Description
Fix generation config save.
use_unified_checkpoint=False
, the model is saved bymodel.save_from_pretrained
, in which thegeneration_config.json
is saved.use_unified_checkpoint=False
, the model is saved by UC, which ignores savinggeneration_config.json
.