kernels-community
/

quantization-bitsandbytes

kernels-bot commited on 2 days ago

Commit

1c312ac

verified ·

1 Parent(s): a3eb31a

Uploaded using `kernel-builder`.

Files changed (1) hide show

README.md CHANGED Viewed

@@ -11,7 +11,7 @@ This is the repository card of kernels-community/quantization-bitsandbytes that
 # make sure `kernels` is installed: `pip install -U kernels`
 from kernels import get_kernel
-kernel_module = get_kernel("kernels-community/quantization-bitsandbytes")
 gemm_4bit_forward = kernel_module.gemm_4bit_forward
 gemm_4bit_forward(...)

 # make sure `kernels` is installed: `pip install -U kernels`
 from kernels import get_kernel
+kernel_module = get_kernel("kernels-community/quantization-bitsandbytes", version=1)
 gemm_4bit_forward = kernel_module.gemm_4bit_forward
 gemm_4bit_forward(...)