diff --git a/bitsandbytes/optim/lion.py b/bitsandbytes/optim/lion.py index c267af7..2551b68 100644 --- a/bitsandbytes/optim/lion.py +++ b/bitsandbytes/optim/lion.py @@ -22,7 +22,7 @@ class Lion(Optimizer1State): "lion", params, lr, - (beta1, beta2), + betas, 0., weight_decay, optim_bits, @@ -49,7 +49,7 @@ class Lion8bit(Optimizer1State): "lion", params, lr, - (beta1, beta2), + betas, 0., weight_decay, 8,