Fix typos (#33583)

Co-authored-by: litianjian <litianjian@bytedance.com>
huggingface · Sep 20, 2024 · 7b2b536 · 7b2b536
1 parent e9356a4
commit 7b2b536
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/transformers/models/llava_onevision/modeling_llava_onevision.py b/src/transformers/models/llava_onevision/modeling_llava_onevision.py
@@ -475,8 +475,8 @@ def apply_pooling(self, image_features):
         image_features = image_features.view(batch_frames, height, width, -1)
         image_features = image_features.permute(0, 3, 1, 2).contiguous()
 
-        height, weight = image_features.shape[2:]
-        scaled_shape = [math.ceil(height / 2), math.ceil(weight / 2)]
+        height, width = image_features.shape[2:]
+        scaled_shape = [math.ceil(height / 2), math.ceil(width / 2)]
         image_features = nn.functional.interpolate(image_features, size=scaled_shape, mode="bilinear")
 
         image_features = image_features.permute(0, 2, 3, 1)