Skip to content

Commit ac67fcc

Browse files
author
sangchengmeng
committed
fix other vlm
1 parent b477506 commit ac67fcc

File tree

5 files changed

+3
-5
lines changed

5 files changed

+3
-5
lines changed

lightllm/models/gemma3/gemma3_visual.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616

1717

1818
class Gemma3VisionModel:
19-
def __init__(self, kvargs):
19+
def __init__(self):
2020
pass
2121

2222
def load_model(self, weight_dir):

lightllm/models/llava/llava_visual.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515

1616

1717
class LlavaVisionModel:
18-
def __init__(self, kvargs):
18+
def __init__(self):
1919
pass
2020

2121
def load_model(self, weight_dir):

lightllm/models/qwen_vl/qwen_visual.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,7 @@ def encode(self, image_uuids: List):
423423
for i, item in enumerate(image_uuids):
424424
if isinstance(item, int):
425425
uuids.append(item)
426-
image_data = read_shm(get_shm_name_data(item.uuid))
426+
image_data = read_shm(get_shm_name_data(item))
427427
image_data = Image.open(BytesIO(image_data)).convert("RGB")
428428
t = self.image_transform(image_data)
429429
img_tensors.append(t)

lightllm/models/tarsier2/tarsier2_visual.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,6 @@ def forward(self, image_features, input_embeddings):
152152
class TarsierVisionTransformerPretrainedModel(nn.Module):
153153
def __init__(
154154
self,
155-
kvargs,
156155
vision_config=None,
157156
text_config=None,
158157
ignore_index=-100,

lightllm/models/vit/model.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import os
2-
import time
32
import json
43
import torch
54
from lightllm.models.vit.layer_infer.pre_layer_infer import ViTPreLayerInfer

0 commit comments

Comments
 (0)