Update fast_lora.py

This commit is contained in:
Daniel Han-Chen 2024-01-26 16:43:50 +11:00
commit 7bed11a8ba

View file

@ -130,7 +130,6 @@ class LoRA_MLP(torch.autograd.Function):
df = se * (1 - f) + f
DW_f = DW * f
DW_dfg = DW * df * g
h, DW_f, DW_dfg = DW, e, 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