mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

whoops

Browse Source
This commit is contained in:
Phil Wang 2023-03-10 08:59:49 -08:00
parent c99b44f774
commit 19b9ef34b9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bitsandbytes/optim/lion.py
View File

@ -22,7 +22,7 @@ class Lion(Optimizer1State):
"lion", "lion",
params, params,
lr, lr,
(beta1, beta2), betas,
0., 0.,
weight_decay, weight_decay,
optim_bits, optim_bits,
@ -49,7 +49,7 @@ class Lion8bit(Optimizer1State):
"lion", "lion",
params, params,
lr, lr,
(beta1, beta2), betas,
0., 0.,
weight_decay, weight_decay,
8, 8,