Skip to content

GitLab

Unverified Commit a9a1c44a authored by EtienneDosSantos's avatar EtienneDosSantos Committed by GitHub
Browse files

Add `"lamb"` to `str2optimizer32bit`

parent 79815ad4
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
bitsandbytes/functional.py
View file @ a9a1c44a
...@@ -49,6 +49,10 @@ if lib and lib.compiled_with_cuda: ...@@ -49,6 +49,10 @@ if lib and lib.compiled_with_cuda:
lib.cadagrad32bit_grad_32, lib.cadagrad32bit_grad_32,
lib.cadagrad32bit_grad_16, lib.cadagrad32bit_grad_16,
), ),
"lamb": (
lib.cadam32bit_grad_fp32,
lib.cadam32bit_grad_fp16,
),
} }
str2optimizer8bit = { str2optimizer8bit = {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment