From 050c61bc0b77414bf9e26828786c2c81f841ff62 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Mon, 22 Jan 2024 20:04:27 +1100 Subject: [PATCH] Update utils.py --- unsloth/kernels/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/kernels/utils.py b/unsloth/kernels/utils.py index 3bdeaebe9a..141814c332 100644 --- a/unsloth/kernels/utils.py +++ b/unsloth/kernels/utils.py @@ -177,7 +177,7 @@ def fast_gemv(X, W, quant_state, out = None, out_W = None): pass -def fast_linear_forward(proj, X, temp_lora = None, out = None) +def fast_linear_forward(proj, X, temp_lora = None, out = None): W, W_quant, lora_A, lora_B, lora_S = get_lora_parameters(proj) out = fast_gemv(X, W, W_quant, out = out) if lora_A is not None: