Skip to content

Commit ad1f503

Browse files
committed
del ov_pipe
1 parent bde70d2 commit ad1f503

File tree

1 file changed

+1
-14
lines changed

1 file changed

+1
-14
lines changed

tests/python_tests/test_vlm_pipeline.py

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -263,25 +263,12 @@ def test_vlm_continuous_batching_generate_vs_add_request(config, cat_tensor):
263263
def test_vlm_continuous_batching_generate_vs_add_request_for_gemma(config, cat_tensor):
264264
scheduler_config = SchedulerConfig()
265265
models_path = get_ov_model(model_ids[8])
266-
ov_pipe = VLMPipeline(
267-
models_path,
268-
"CPU",
269-
scheduler_config=scheduler_config,
270-
**get_default_llm_properties(),
271-
)
266+
272267
generation_config = config
273268
generation_config.max_new_tokens = 30
274269
eps = 0.001
275270
image_links_list = [[], [cat_tensor]]
276271

277-
res_generate = []
278-
for images in image_links_list:
279-
res_generate.append(
280-
ov_pipe.generate(
281-
prompts[0], images=images, generation_config=generation_config
282-
)
283-
)
284-
285272
cb_pipe = ContinuousBatchingPipeline(
286273
models_path,
287274
scheduler_config=scheduler_config,

0 commit comments

Comments
 (0)