Skip to content

Commit 513e198

Browse files
committed
update the keyword for matching key_value_input_names
1 parent 25c6976 commit 513e198

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

onnxruntime/core/providers/openvino/ov_stateful_patch_utils.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,10 +144,10 @@ void PatchStatefulDecoder(std::shared_ptr<ov::Model> model) {
144144
if (param_name.find("key_values") != std::string::npos) {
145145
key_value_input_names.push_back(param_name);
146146
found = true;
147-
} else if (param_name.find("key") != std::string::npos) {
147+
} else if (param_name.find("keys") != std::string::npos) {
148148
key_value_input_names.push_back(param_name);
149149
found = true;
150-
} else if (param_name.find("value") != std::string::npos) {
150+
} else if (param_name.find("values") != std::string::npos) {
151151
key_value_input_names.push_back(param_name);
152152
found = true;
153153
}

0 commit comments

Comments
 (0)