File tree 1 file changed +8
-6
lines changed
vllm/model_executor/models
1 file changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -87,17 +87,19 @@ def _get_llava_next_num_unpadded_features(
87
87
current_height = npatches * num_patch_height
88
88
current_width = npatches * num_patch_width
89
89
90
- aspect_ratio = original_width / original_height
90
+ original_aspect_ratio = original_width / original_height
91
91
current_aspect_ratio = current_width / current_height
92
92
93
- if aspect_ratio > current_aspect_ratio :
94
- new_height = (original_height * current_width ) // original_width
93
+ if original_aspect_ratio > current_aspect_ratio :
94
+ scale_factor = current_width / original_width
95
+ new_height = int (original_height * scale_factor )
95
96
padding = (current_height - new_height ) // 2
96
- current_height -= padding * 2
97
+ current_height -= 2 * padding
97
98
else :
98
- new_width = (original_width * current_height ) // original_height
99
+ scale_factor = current_height / original_height
100
+ new_width = int (original_width * scale_factor )
99
101
padding = (current_width - new_width ) // 2
100
- current_width -= padding * 2
102
+ current_width -= 2 * padding
101
103
102
104
unpadded_features = current_height * current_width
103
105
newline_features = current_height
You can’t perform that action at this time.
0 commit comments