20231088/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[Bugfix] Fix precommit - line too long in pixtral.py (#14960)

Browse Source 
Signed-off-by: Tyler Michael Smith <tyler@neuralmagic.com>
Co-authored-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Tyler Michael Smith 2025-03-17 10:18:50 -04:00 committed by GitHub
parent 89fca671fb
commit e1eb45d397
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements/test.txt
View File

@ -235,7 +235,7 @@ mbstrdecoder==1.1.3
# typepy # typepy
mdurl==0.1.2 mdurl==0.1.2
# via markdown-it-py # via markdown-it-py
mistral-common==1.5.1 mistral-common==1.5.4
# via -r requirements/test.in # via -r requirements/test.in
more-itertools==10.5.0 more-itertools==10.5.0
# via lm-eval # via lm-eval

3
vllm/model_executor/models/pixtral.py
View File

@ -935,7 +935,8 @@ class PatchMerger(nn.Module):
# x is (N, vision_encoder_dim) # x is (N, vision_encoder_dim)
x = self.permute(x, image_sizes) x = self.permute(x, image_sizes)
# x is (N / spatial_merge_size ** 2, vision_encoder_dim * spatial_merge_size ** 2) # x is (N / spatial_merge_size ** 2,
# vision_encoder_dim * spatial_merge_size ** 2)
x = self.merging_layer(x) x = self.merging_layer(x)
# x is (N / spatial_merge_size ** 2, vision_encoder_dim) # x is (N / spatial_merge_size ** 2, vision_encoder_dim)