Update llama.py
This commit is contained in:
parent
9df19aeb9f
commit
1b274212e1
1 changed files with 1 additions and 1 deletions
|
|
@ -699,7 +699,7 @@ def LlamaModel_fast_forward_inference(
|
|||
decoder_layer.self_attn,
|
||||
hidden_states,
|
||||
past_key_values[idx],
|
||||
position_ids,
|
||||
None,
|
||||
)
|
||||
hidden_states += residual
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue