[bug] Fix set tensor name for `attention_mask` #352

praasz · 2024-12-16T13:07:33Z

Fix set attention_mask tensor name as model output tensor name.
When is required that tensor name(s) should be model output tensor names they should be set to Result` node or OV Model output tensor. The OV pre-post-processor for model v11 will not move the tensor names. If tensor names is not assigned to Result then it will be kept on that node when additional layer(s) added between Node and Result.

Signed-off-by: Raasz, Pawel <[email protected]>

### Description: - Fix the GENAI test after PPP don't move Node's name and tensor names. ### Related PRs: - openvinotoolkit/openvino_tokenizers#352 - openvinotoolkit/openvino#28069 --------- Signed-off-by: Raasz, Pawel <[email protected]>

Fix set tensor name for attention_mask

d24b4ba

Signed-off-by: Raasz, Pawel <[email protected]>

praasz requested review from pavel-esir and mryzhov December 16, 2024 13:07

github-actions bot added the tokenizers_py label Dec 16, 2024

praasz added this to the 2025.0 milestone Dec 16, 2024

praasz mentioned this pull request Dec 16, 2024

Fix test fails after PPP stops move tensor names openvinotoolkit/openvino.genai#1390

Merged

ilya-lavrenov assigned pavel-esir and apaniukov Dec 16, 2024

pavel-esir approved these changes Dec 16, 2024

View reviewed changes

ilya-lavrenov added this pull request to the merge queue Dec 16, 2024

apaniukov approved these changes Dec 16, 2024

View reviewed changes

Merged via the queue into openvinotoolkit:master with commit bcfd3ed Dec 16, 2024
18 checks passed

praasz deleted the bugfix/update-set-attention-mask-tensor-name branch December 17, 2024 05:46

ilya-lavrenov added the bug Something isn't working label Dec 17, 2024

Provide feedback