diff --git a/bitsandbytes/backends/cuda/ops.py b/bitsandbytes/backends/cuda/ops.py index cb2796b1f..ecafbdf25 100644 --- a/bitsandbytes/backends/cuda/ops.py +++ b/bitsandbytes/backends/cuda/ops.py @@ -574,6 +574,10 @@ def _gemv_4bit_impl( lib.cademamix32bit_grad_fp16, lib.cademamix32bit_grad_bf16, ), + "lars": ( + lib.cmomentum32bit_grad_32, + lib.cmomentum32bit_grad_16, + ), } str2optimizer8bit_blockwise = {