Update __init__.py

This commit is contained in:
Daniel Han-Chen 2024-01-30 03:23:42 +11:00
commit 71725aeea5

View file

@ -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