"TypeError: get_cosine_schedule_with_warmup() got an unexpected keyword argument 'num_decay_steps'" Error occurs in dev branch

[tatamizu]

When I run training on the dev branch, I get this error

Traceback (most recent call last):
File "G:\SD_Training\dev_version\sd-scripts\sdxl_train_network.py", line 185, in
trainer.train(args)
File "G:\SD_Training\dev_version\sd-scripts\train_network.py", line 395, in train
lr_scheduler = train_util.get_scheduler_fix(args, optimizer, accelerator.num_processes)
File "G:\SD_Training\dev_version\sd-scripts\library\train_util.py", line 4513, in get_scheduler_fix
return schedule_func(
TypeError: get_cosine_schedule_with_warmup() got an unexpected keyword argument 'num_decay_steps'

Reinstalling did not solve the problem.

[kohya-ss]

Sorry, I've fixed this on dev branch too.