fast break

openvinotoolkit · Nov 15, 2024 · 541b714 · 541b714
1 parent b6dd010
commit 541b714
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/core/src/pass/sdpa_to_paged_attention.cpp b/src/core/src/pass/sdpa_to_paged_attention.cpp
@@ -73,7 +73,9 @@ bool ov::pass::SDPAToPagedAttention::run_on_model(const std::shared_ptr<ov::Mode
 
     std::shared_ptr<v0::Parameter> input_ids_node;
     for (const auto& name : {"input_ids", "inputs_embeds"}) {
-        input_ids_node = get_parameter(model, name);
+        if (input_ids_node = get_parameter(model, name)) {
+            break;
+        }
     }
 
     if (!input_ids_node) {