-
Notifications
You must be signed in to change notification settings - Fork 606
Support InternVL3.5-Flash #3952
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
Conversation
lmdeploy/pytorch/models/internvl.py
Outdated
|
||
def mlp_block(in_dim, out_dim): | ||
return nn.Sequential( | ||
nn.Linear(in_dim, out_dim), |
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.
Any reason for not using build_rowwise_linear
and build_colwise_linear
?
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.
https://github.com/InternLM/lmdeploy/blob/main/lmdeploy/pytorch/models/internvl.py#L368
We can. But similar to the above code, as the vision encoding is only executed for once, and also for simplicity, I do not replace it with a custom op.
If you think it's worthwhile to do, I will replace them
@grimoire may resolve the following warning in PR #3922
|
Performance & Acc
Test with VLMEvalKit, dataset: ORCBench
InternVL3.5-FLash acc details
Related