[Feature] VLM model support tp #206
Draft
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.
Motivation
support tp for qwen2.5 vl, the gpu memory is 78.62GB in tp1, 43.56GB in tp4.
Modifications
add qwen2_5_vl.py for target model
add
QKVParallelLinear
for linear.py, becauseQwen2_5_VLVisionAttention
class need it.Related Issues
#166
Pedding todo
accuracy test
support tp8, because
num_attention_heads
in config.json can not be divide by 8.Checklist