Update llama.py

This commit is contained in:
Daniel Han-Chen 2024-01-28 04:08:19 +11:00
commit 6c7f0dbcb4

View file

@ -487,7 +487,6 @@ def LlamaModel_fast_forward(
inputs_embeds = self.embed_tokens(input_ids)
# Ignore attention_mask
print(attention_mask)
if attention_mask is None:
padding_mask = None
elif False:#self.training: