Update llama.py

This commit is contained in:
Daniel Han-Chen 2024-02-02 20:24:28 +11:00
commit 9146fa443d

View file

@ -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,