Skip to content

Commit

Permalink
update requirements.txt and README to include RAdamScheduleFree optim…
Browse files Browse the repository at this point in the history
…izer support
  • Loading branch information
kohya-ss committed Dec 15, 2024
1 parent f2d38e6 commit e896539
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 1 deletion.
6 changes: 6 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,12 @@ The command to install PyTorch is as follows:

### Recent Updates

Dec 15, 2024:

- RAdamScheduleFree optimizer is supported. PR [#1830](https://github.com/kohya-ss/sd-scripts/pull/1830) Thanks to nhamanasu!
- Update to `schedulefree==1.4` is required. Please update individually or with `pip install --use-pep517 --upgrade -r requirements.txt`.
- Available with `--optimizer_type=RAdamScheduleFree`. No need to specify warm up steps as well as learning rate scheduler.

Dec 7, 2024:

- The option to specify the model name during ControlNet training was different in each script. It has been unified. Please specify `--controlnet_model_name_or_path`. PR [#1821](https://github.com/kohya-ss/sd-scripts/pull/1821) Thanks to sdbds!
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ pytorch-lightning==1.9.0
bitsandbytes==0.44.0
prodigyopt==1.0
lion-pytorch==0.0.6
schedulefree==1.2.7
schedulefree==1.4
tensorboard
safetensors==0.4.4
# gradio==3.16.2
Expand Down

0 comments on commit e896539

Please sign in to comment.