Skip to content

Commit 7d5b9a6

Browse files
author
sangchengmeng
committed
fix other vlm
1 parent d53a924 commit 7d5b9a6

File tree

4 files changed

+4
-16
lines changed

4 files changed

+4
-16
lines changed

lightllm/models/llava/llava_visual.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -134,10 +134,7 @@ def encode(self, images: List[ImageItem]):
134134
for i, img in enumerate(images):
135135
if isinstance(img, ImageItem):
136136
uuids.append(img.uuid)
137-
if self.remote_vit:
138-
image_data = img._preload_data
139-
else:
140-
image_data = read_shm(get_shm_name_data(img.uuid))
137+
image_data = read_shm(get_shm_name_data(img.uuid))
141138
image_data = Image.open(BytesIO(image_data)).convert("RGB")
142139
t = self.image_processor.preprocess(image_data, return_tensors="pt")["pixel_values"]
143140
img_tensors.append(t)

lightllm/models/qwen2_5_vl/qwen2_5_visual.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -382,10 +382,7 @@ def encode(self, images: List[ImageItem]):
382382
for i, img in enumerate(images):
383383
if isinstance(img, ImageItem):
384384
uuids.append(img.uuid)
385-
if self.remote_vit:
386-
image_data = img._preload_data
387-
else:
388-
image_data = read_shm(get_shm_name_data(img.uuid))
385+
image_data = read_shm(get_shm_name_data(img.uuid))
389386
image_data = Image.open(BytesIO(image_data))
390387
image_data = resize_image(image_data)
391388
pixel_values, image_grid_thw = self.processor.preprocess(image_data)

lightllm/models/qwen2_vl/qwen2_visual.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -310,10 +310,7 @@ def encode(self, images: List[ImageItem]):
310310
for i, img in enumerate(images):
311311
if isinstance(img, ImageItem):
312312
uuids.append(img.uuid)
313-
if self.remote_vit:
314-
image_data = img._preload_data
315-
else:
316-
image_data = read_shm(get_shm_name_data(img.uuid))
313+
image_data = read_shm(get_shm_name_data(img.uuid))
317314
image_data = Image.open(BytesIO(image_data))
318315
image_data = resize_image(image_data)
319316
pixel_values, image_grid_thw = self.processor.preprocess(image_data)

lightllm/models/tarsier2/tarsier2_visual.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -253,10 +253,7 @@ def encode(self, images: List[ImageItem]):
253253
for i, img in enumerate(images):
254254
if isinstance(img, ImageItem):
255255
uuids.append(img.uuid)
256-
if self.remote_vit:
257-
image_data = img._preload_data
258-
else:
259-
image_data = read_shm(get_shm_name_data(img.uuid))
256+
image_data = read_shm(get_shm_name_data(img.uuid))
260257
image_data = Image.open(BytesIO(image_data))
261258
image_data = resize_image(image_data)
262259
pixel_values, image_grid_thw = self.processor.preprocess(image=image_data)

0 commit comments

Comments
 (0)