Skip to content

Commit

Permalink
šŸ› fix(llava_onevision): update pretrained model and conversation tempā€¦
Browse files Browse the repository at this point in the history
ā€¦late (#478)
  • Loading branch information
pufanyi authored and kcz358 committed Dec 30, 2024
1 parent 9823c33 commit 7d402cb
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions lmms_eval/models/llava_onevision.py
Original file line number Diff line number Diff line change
Expand Up @@ -69,14 +69,14 @@ class Llava_OneVision(lmms):

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

0 comments on commit 7d402cb

Please sign in to comment.