From a922dab70fe93eb2778258edd71669d726b1378c Mon Sep 17 00:00:00 2001
From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
Date: Mon, 16 Sep 2024 11:06:57 -0400
Subject: [PATCH] Add PagedAdEMAMix32bit, AdEMAMix32bit

---
 bitsandbytes/optim/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bitsandbytes/optim/__init__.py b/bitsandbytes/optim/__init__.py
index ea2cf5b15..07174c38d 100644
--- 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