Add PagedAdEMAMix32bit, AdEMAMix32bit

bitsandbytes-foundation · Sep 16, 2024 · a922dab · a922dab
1 parent 0911854
commit a922dab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bitsandbytes/optim/__init__.py b/bitsandbytes/optim/__init__.py
@@ -13,7 +13,7 @@
     PagedAdamW8bit,
     PagedAdamW32bit,
 )
-from .ademamix import AdEMAMix, AdEMAMix8bit, PagedAdEMAMix, PagedAdEMAMix8bit
+from .ademamix import AdEMAMix, AdEMAMix8bit, AdEMAMix32bit, PagedAdEMAMix, PagedAdEMAMix8bit, PagedAdEMAMix32bit
 from .lamb import LAMB, LAMB8bit, LAMB32bit
 from .lars import LARS, LARS8bit, LARS32bit, PytorchLARS
 from .lion import Lion, Lion8bit, Lion32bit, PagedLion, PagedLion8bit, PagedLion32bit