diff --git a/requirements.txt b/requirements.txt index 624978b4..ce1ebd8f 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,7 +9,7 @@ einops==0.7.0 bitsandbytes lion-pytorch==0.2.3 schedulefree==1.4 -pytorch-optimizer==3.7.0 +pytorch-optimizer==3.9.0 prodigy-plus-schedule-free==1.9.2 prodigyopt==1.1.2 tensorboard