Skip to content

Commit 3c00482

Browse files
authored
šŸ› fix(llava_onevision): update pretrained model and conversation template (#478)
1 parent 7f9c482 commit 3c00482

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ā€Žlmms_eval/models/llava_onevision.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,14 +69,14 @@ class Llava_OneVision(lmms):
6969

7070
def __init__(
7171
self,
72-
pretrained: str = "liuhaotian/llava-v1.5-7b",
72+
pretrained: str = "lmms-lab/llava-onevision-qwen2-7b-ov",
7373
truncation: Optional[bool] = True,
7474
device: Optional[str] = "cuda:0",
7575
batch_size: Optional[Union[int, str]] = 1,
7676
model_name: Optional[str] = None,
7777
attn_implementation: Optional[str] = best_fit_attn_implementation,
7878
device_map: Optional[str] = "cuda:0",
79-
conv_template: Optional[str] = "vicuna_v1",
79+
conv_template: Optional[str] = "qwen_1_5",
8080
use_cache: Optional[bool] = True,
8181
truncate_context: Optional[bool] = False, # whether to truncate the context in generation, set it False for LLaVA-1.6
8282
customized_config: Optional[str] = None, # ends in json

0 commit comments

Comments
Ā (0)