Update utils.py
This commit is contained in:
parent
1c7ecc83a7
commit
d36dbaed87
1 changed files with 1 additions and 1 deletions
|
|
@ -438,7 +438,7 @@ def fast_linear_forward(proj, X, temp_lora = None, out = None):
|
|||
elif bsz == 1 and q_len == 1:
|
||||
out = fast_gemv(X, W, W_quant, out = out)
|
||||
else:
|
||||
W = fast_dequantize(W.t(), W_quant, use_global_buffer = True)
|
||||
W = fast_dequantize(W.t(), W_quant, use_global_buffer = False)
|
||||
out = torch_matmul(X, W, out = out)
|
||||
pass
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue