Skip to content

Commit d151833

Browse files
committed
docs: update README with recent changes and specify version for pytorch-optimizer
1 parent 936d333 commit d151833

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,10 @@ The command to install PyTorch is as follows:
1414

1515
### Recent Updates
1616

17+
Mar 20, 2025:
18+
- `pytorch-optimizer` is added to requirements.txt. Thank you to gesen2egee for PR [#1985](https://github.com/kohya-ss/sd-scripts/pull/1985).
19+
- For example, you can use CAME optimizer with `--optimizer_type "pytorch_optimizer.CAME" --optimizer_args "weight_decay=0.01"`.
20+
1721
Mar 6, 2025:
1822

1923
- Added a utility script to merge the weights of SD3's DiT, VAE (optional), CLIP-L, CLIP-G, and T5XXL into a single .safetensors file. Run `tools/merge_sd3_safetensors.py`. See `--help` for usage. PR [#1960](https://github.com/kohya-ss/sd-scripts/pull/1960)

requirements.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,9 @@ pytorch-lightning==1.9.0
99
bitsandbytes==0.44.0
1010
lion-pytorch==0.0.6
1111
schedulefree==1.4
12+
pytorch-optimizer==3.5.0
13+
prodigy-plus-schedule-free==1.9.0
14+
prodigyopt==1.1.2
1215
tensorboard
1316
safetensors==0.4.4
1417
# gradio==3.16.2
@@ -42,7 +45,4 @@ rich==13.7.0
4245
# for T5XXL tokenizer (SD3/FLUX)
4346
sentencepiece==0.2.0
4447
# for kohya_ss library
45-
pytorch-optimizer
46-
prodigy-plus-schedule-free==1.9.0
47-
prodigyopt==1.1.2
4848
-e .

0 commit comments

Comments
 (0)