fix llava new processing test

IlyasMoutawwakil · IlyasMoutawwakil · commit f5b30f28234d · 2025-02-04T09:03:39.000+01:00
diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -2237,6 +2237,7 @@ def test_llava_with_new_preprocessing(self, model_arch):
             patch_size=config.vision_config.patch_size,
             vision_feature_select_strategy=config.vision_feature_select_strategy,
             trust_remote_code=model_arch in self.REMOTE_CODE_MODELS,
+            num_additional_image_tokens=1,
         )
         transformers_model = self.get_transformer_model_class(model_arch).from_pretrained(model_id)
         ov_model = OVModelForVisualCausalLM.from_pretrained(

Original file line number	Diff line number	Diff line change
`@@ -2237,6 +2237,7 @@ def test_llava_with_new_preprocessing(self, model_arch):`
`2237`	`2237`	`patch_size=config.vision_config.patch_size,`
`2238`	`2238`	`vision_feature_select_strategy=config.vision_feature_select_strategy,`
`2239`	`2239`	`trust_remote_code=model_arch in self.REMOTE_CODE_MODELS,`
	`2240`	`+ num_additional_image_tokens=1,`
`2240`	`2241`	`)`
`2241`	`2242`	`transformers_model = self.get_transformer_model_class(model_arch).from_pretrained(model_id)`
`2242`	`2243`	`ov_model = OVModelForVisualCausalLM.from_pretrained(`