Skip to content

Commit 3d112ac

Browse files
authored
fix
1 parent 039fe09 commit 3d112ac

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lightllm/models/qwen2_vl/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,8 @@ def get_image_token_length(self, img: ImageItem):
5555
height=height, width=width, min_pixels=self.min_pixel, max_pixels=self.max_pixel
5656
)
5757
grid_h, grid_w = resized_height // self.patch_size, resized_width // self.patch_size
58-
self.token_num = (grid_h * grid_w) // (self.merge_size ** 2)
59-
return self.token_num
58+
token_num = (grid_h * grid_w) // (self.merge_size ** 2)
59+
return token_num
6060

6161
def get_audio_token_length(self, audio: AudioItem):
6262
raise NotImplementedError

0 commit comments

Comments
 (0)