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-26 20:04:51 +11:00
commit 4f573494da
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
unsloth/kernels/fast_lora.py
View file

@ -124,11 +124,14 @@ class LoRA_MLP(torch.autograd.Function):
DW = matmul_lora(dY, downW.t(), downW_quant, downB, downA, downS)
se = torch.nn.functional.sigmoid(e)
f = e * se
h = f * g
df = se * (1 - f) + f
f = torch.nn.functional.silu(e)
DW_f = DW * f
DW_dfg = DW * df * g
DW_dfg = DW * se * g * (1 + f * (1 - se))
# f = e * se
# h = f * g
# df = se * (1 - f) + f
# DW_f = DW * f
# DW_dfg = DW * df * g
# DW = matmul_lora(dY, downW.t(), downW_quant, downB, downA, downS)
# DW, e, g = swiglu_DWf_DW_dfg_kernel(DW, e, g)
# h, DW_f, DW_dfg = DW, e, g