Add PagedAdamW32bit to Dropdown

bmaltais · Nov 18, 2023 · 0c0827f · 0c0827f
1 parent 17f58c8
commit 0c0827f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/library/class_basic_training.py b/library/class_basic_training.py
@@ -105,6 +105,7 @@ def __init__(
                     "Lion",
                     "Lion8bit",
                     "PagedAdamW8bit",
+                    "PagedAdamW32bit",
                     "PagedLion8bit",
                     "Prodigy",
                     "SGDNesterov",