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 2025-03-21 17:00:35 -07:00
commit df44d6a4fb
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

@ -654,7 +654,7 @@ def LlamaModel_fast_forward(
inputs_embeds = self.embed_tokens(input_ids)
inputs_embeds = inputs_embeds.to(_get_dtype(self.config.torch_dtype))
print(inputs_embeds.shape)
print(inputs_embeds.shape, input_ids.shape, input_ids)
# Normalized from Gemma
IS_GEMMA = self.config.model_type.startswith("gemma")