diff --git a/kohya_gui/class_basic_training.py b/kohya_gui/class_basic_training.py index 2bf6ff9a..84a42902 100644 --- a/kohya_gui/class_basic_training.py +++ b/kohya_gui/class_basic_training.py @@ -212,6 +212,7 @@ def init_lr_and_optimizer_controls(self) -> None: "PagedAdamW32bit", "PagedLion8bit", "Prodigy", + "prodigyplus.ProdigyPlusScheduleFree", "RAdamScheduleFree", "SGDNesterov", "SGDNesterov8bit", diff --git a/requirements.txt b/requirements.txt index 728c0e44..a5234ec1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -20,6 +20,7 @@ protobuf==3.20.3 open-clip-torch==2.20.0 opencv-python==4.10.0.84 prodigyopt==1.0 +prodigy-plus-schedule-free==1.8.0 pytorch-lightning==1.9.0 rich>=13.7.1 safetensors==0.4.4