From 71725aeea568b7841282736855971ad71efbc0e3 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Tue, 30 Jan 2024 03:23:42 +1100 Subject: [PATCH] Update __init__.py --- unsloth/kernels/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/kernels/__init__.py b/unsloth/kernels/__init__.py index 9b861f4934..f5db8fa890 100644 --- a/unsloth/kernels/__init__.py +++ b/unsloth/kernels/__init__.py @@ -22,4 +22,4 @@ from .fast_lora import ( apply_lora_qkv, apply_lora_o, ) -from .utils import fast_dequantize, QUANT_STATE, fast_linear_forward +from .utils import fast_dequantize, fast_gemv, QUANT_STATE, fast_linear_forward