7bc93821a0
- Caption dropout is supported in ``train_db.py``, ``fine_tune.py`` and ``train_network.py``. Thanks to forestsource!
- ``--caption_dropout_rate`` option specifies the dropout rate for captions (0~1.0, 0.1 means 10% chance for dropout). If dropout occurs, the image is trained with the empty caption. Default is 0 (no dropout).
- ``--caption_dropout_every_n_epochs`` option specifies how many epochs to drop captions. If ``3`` is specified, in epoch 3, 6, 9 ..., images are trained with all captions empty. Default is None (no dropout).
- ``--caption_tag_dropout_rate`` option specified the dropout rate for tags (comma separated tokens) (0~1.0, 0.1 means 10% chance for dropout). If dropout occurs, the tag is removed from the caption. If ``--keep_tokens`` option is set, these tokens (tags) are not dropped. Default is 0 (no droupout).
- The bulk image downsampling script is added. Documentation is [here](https://github.com/kohya-ss/sd-scripts/blob/main/train_network_README-ja.md#%E7%94%BB%E5%83%8F%E3%83%AA%E3%82%B5%E3%82%A4%E3%82%BA%E3%82%B9%E3%82%AF%E3%83%AA%E3%83%97%E3%83%88) (in Jpanaese). Thanks to bmaltais!
- Typo check is added. Thanks to shirayu!
- Add option to autolaunch the GUI in a browser and set the server_port. USe either `gui.ps1 --inbrowser --server_port 3456`or `gui.cmd -inbrowser -server_port 3456`
26 lines
589 B
Plaintext
26 lines
589 B
Plaintext
accelerate==0.15.0
|
|
transformers==4.26.0
|
|
ftfy==6.1.1
|
|
albumentations==1.3.0
|
|
opencv-python==4.7.0.68
|
|
einops==0.6.0
|
|
diffusers[torch]==0.10.2
|
|
pytorch-lightning==1.9.0
|
|
bitsandbytes==0.35.0
|
|
tensorboard==2.10.1
|
|
safetensors==0.2.6
|
|
gradio==3.16.2
|
|
altair==4.2.2
|
|
easygui==0.98.3
|
|
tk==0.1.0
|
|
# for BLIP captioning
|
|
requests==2.28.2
|
|
timm==0.6.12
|
|
fairscale==0.4.13
|
|
# for WD14 captioning
|
|
# tensorflow<2.11
|
|
tensorflow==2.10.1
|
|
huggingface-hub==0.12.0
|
|
xformers @ https://github.com/C43H66N12O12S2/stable-diffusion-webui/releases/download/f/xformers-0.0.14.dev0-cp310-cp310-win_amd64.whl
|
|
# for kohya_ss library
|
|
. |