Update llama.py
This commit is contained in:
parent
a5a123db08
commit
9146fa443d
1 changed files with 0 additions and 1 deletions
|
|
@ -680,7 +680,6 @@ pass
|
|||
|
||||
# https://github.com/huggingface/transformers/blob/main/src/transformers/models/llama/modeling_llama.py#L825
|
||||
@torch.compile
|
||||
@torch.inference_mode
|
||||
def LlamaModel_fast_forward_inference(
|
||||
self,
|
||||
input_ids,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue