Throw Exception for Unexpected Type (openvinotoolkit#851)

apaniukov · Jan 25, 2024 · 2e0bd47 · 2e0bd47
1 parent 71a1d2b
commit 2e0bd47
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/modules/custom_operations/user_ie_extensions/tokenizer/sentence_piece.cpp b/modules/custom_operations/user_ie_extensions/tokenizer/sentence_piece.cpp
@@ -159,6 +159,8 @@ bool SentencepieceTokenizer::evaluate(TensorVector& outputs, const TensorVector&
         batch_size = static_cast<int32_t>(ov::shape_size(inputs[1].get_shape()));
     } else if(input_element_type == ov::element::u8) {
         parse_packed_strings(inputs[1], batch_size, begin_ids, end_ids, data);
+    } else {
+        OPENVINO_THROW("Unexpected input type during inference. SentencepieceTokenizer accepts element::u8 or element::string.");
     }
 
 #endif