vrr/unsloth
2
0
Fork 0
mirror of https://github.com/unslothai/unsloth.git synced 2026-07-10 00:08:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 29

Update fast_lora.py

Browse source
This commit is contained in:
Daniel Han-Chen 2023-11-30 23:07:19 +11:00
parent 5724e8e389
commit dfe2e406b2
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/kernels/fast_lora.py
View file

@ -43,7 +43,7 @@ def matmul_lora(X, W, W_quant, A, B, s, out = None):
out = torch.matmul(X, W, out = out)
if W_quant is not None: del W
out += (X @ A) @ (s * B)
out += (X @ A.to(dtype)) @ (s * B.to(dtype))
return out.view(batch, seq_len, -1) if reshape else out
pass