File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -69,14 +69,14 @@ class Llava_OneVision(lmms):
69
69
70
70
def __init__ (
71
71
self ,
72
- pretrained : str = "liuhaotian /llava-v1.5-7b " ,
72
+ pretrained : str = "lmms-lab /llava-onevision-qwen2-7b-ov " ,
73
73
truncation : Optional [bool ] = True ,
74
74
device : Optional [str ] = "cuda:0" ,
75
75
batch_size : Optional [Union [int , str ]] = 1 ,
76
76
model_name : Optional [str ] = None ,
77
77
attn_implementation : Optional [str ] = best_fit_attn_implementation ,
78
78
device_map : Optional [str ] = "cuda:0" ,
79
- conv_template : Optional [str ] = "vicuna_v1 " ,
79
+ conv_template : Optional [str ] = "qwen_1_5 " ,
80
80
use_cache : Optional [bool ] = True ,
81
81
truncate_context : Optional [bool ] = False , # whether to truncate the context in generation, set it False for LLaVA-1.6
82
82
customized_config : Optional [str ] = None , # ends in json
You canāt perform that action at this time.
0 commit comments