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

Update fast_lora.py

Browse source
This commit is contained in:
Daniel Han-Chen 2024-02-23 02:38:16 +11:00
commit 6f340f5417
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
unsloth/kernels/fast_lora.py
View file

@ -163,7 +163,8 @@ class LoRA_MLP(torch.autograd.Function):
return dX.view(batch, seq_len, hd), \
None, None, d_gateA.t(), d_gateB.t(), None, \
None, None, d_upA.t(), d_upB.t(), None, \
None, None, d_downA.t(), d_downB.t(), None,
None, None, d_downA.t(), d_downB.t(), None, \
None, None, # _backward and _forward
pass
pass