diff --git a/python/llm/src/ipex_llm/transformers/models/baichuan.py b/python/llm/src/ipex_llm/transformers/models/baichuan.py
index 21d86b06..a4be00ba 100644
--- a/python/llm/src/ipex_llm/transformers/models/baichuan.py
+++ b/python/llm/src/ipex_llm/transformers/models/baichuan.py
@@ -119,7 +119,8 @@ def baichuan_model_7b_forward(
     elif inputs_embeds is not None:
         batch_size, seq_length, _ = inputs_embeds.shape
     else:
-        raise ValueError("You have to specify either decoder_input_ids or decoder_inputs_embeds")
+        log4Error.invalidInputError("You have to specify either decoder_input_ids \
+                                     or decoder_inputs_embeds")
 
     seq_length_with_past = seq_length
     past_key_values_length = 0