From 0c0827ffdb3292a6a3e46a58101e9417c5e900a7 Mon Sep 17 00:00:00 2001 From: xzuyn <16216325+xzuyn@users.noreply.github.com> Date: Sat, 18 Nov 2023 07:22:44 -0500 Subject: [PATCH] Add PagedAdamW32bit to Dropdown --- library/class_basic_training.py | 1 + 1 file changed, 1 insertion(+) diff --git a/library/class_basic_training.py b/library/class_basic_training.py index ac796a345..fe92501ef 100644 --- 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",