version 1.0.5, fix resume checkpoint, model save and generation #148
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
fix lomo with lr_scheduler
srun 命令中加上 --unbuffered 即可显示进度条
…lie into kv_cache_resume_checkpoint
pp时会按照第一个维度切分batch到micro batch，而chatglm2的kvcache第一个维度是seq_len，需要permute一下
之前写法复杂了，实际上generate_batch的输入全是dict就好。返回时候再转换到tuple就行
fix resume checkpoint
解决了tp pp同时使用，保存模型会卡死的问题. gather weight_map in pp group
先tp gather再更换名字和cat query_key_value
update kv cache for generation and fix chatglm2
KaiLv69
changed the title
MorningForest
approved these changes
Jan 3, 2024
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.