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-01-27 01:28:03 +11:00
commit f7760719f0
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/kernels/fast_lora.py
View file

@ -126,8 +126,8 @@ class LoRA_MLP(torch.autograd.Function):
se = 1 / (1 + torch.exp(-e.float()))
f = torch.nn.functional.silu(e)
h = f * g
DW_f = DW * f
DW_dfg = DW.float() * g.float() * se * (1.0 + e.float() - f.float())
DW_f = (DW * f)
DW_dfg = (DW * g).float() * se * (1.0 + e.float() * (1.0 - se))
DW_dfg = DW_dfg.to(dtype)
# f = e * se
# h = f * g