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-02-02 03:56:25 -08:00
commit 8123210d7e
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
unsloth/models/llama.py
View file

@ -936,6 +936,7 @@ def LlamaModel_fast_forward_inference(
next_decoder_cache = []
residual = torch.empty_like(X)
print(bsz, q_len, hd)
_XX = torch.empty((2, bsz, q_len, hd), dtype = torch.float32)
XX, XX2 = _XX[0], _XX[1]
variance = torch.empty((bsz, q_len, 1), dtype = torch.float32, device = "cuda:0")