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",