ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix typos

Browse source
This commit is contained in:
Huang, Xinshengzi 2024-08-22 11:26:49 +08:00
parent 2a0aa9271b
commit c6ed1c412d
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/llm/src/ipex_llm/transformers/models/baichuan.py
View file

@ -119,7 +119,8 @@ def baichuan_model_7b_forward(
elif inputs_embeds is not None: elif inputs_embeds is not None:
batch_size, seq_length, _ = inputs_embeds.shape batch_size, seq_length, _ = inputs_embeds.shape
else: 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 seq_length_with_past = seq_length
past_key_values_length = 0 past_key_values_length = 0