Skip to content

Commit eb55586

Browse files
author
litianjian
committed
Fix typos
1 parent 7125e36 commit eb55586

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

llava/model/llava_arch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -178,8 +178,8 @@ def get_2dPool(self, image_feature, stride=2):
178178
elif self.config.mm_spatial_pool_mode == "max":
179179
image_feature = nn.functional.max_pool2d(image_feature, stride)
180180
elif self.config.mm_spatial_pool_mode == "bilinear":
181-
height, weight = image_feature.shape[2:]
182-
scaled_shape = [math.ceil(height / stride), math.ceil(weight / stride)]
181+
height, width = image_feature.shape[2:]
182+
scaled_shape = [math.ceil(height / stride), math.ceil(width / stride)]
183183
image_feature = nn.functional.interpolate(image_feature, size=scaled_shape, mode='bilinear')
184184

185185
else:

0 commit comments

Comments
 (0)