ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update llama.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-01-28 03:56:54 +11:00
commit 4c5ebcc960
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/llama.py
View file

@ -635,7 +635,7 @@ def LlamaForCausalLM_fast_forward(
logits = self.lm_head(hidden_states)
loss = None
print(labels)
print("labels", labels)
if labels is not None:
shift_logits = logits
if not hasattr(self, "extra_ignored_labels"):