fix typos
This commit is contained in:
parent
2a0aa9271b
commit
c6ed1c412d
1 changed files with 2 additions and 1 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in a new issue