mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-08 14:34:23 +00:00
8abb8645ae
* add config file schema * change config file specification * refactor config utility * unify batch_size to train_batch_size * fix indent size * use batch_size instead of train_batch_size * make cache_latents configurable on subset * rename options * bucket_repo_range * shuffle_keep_tokens * update readme * revert to min_bucket_reso & max_bucket_reso * use subset structure in dataset * format import lines * split mode specific options * use only valid subset * change valid subsets name * manage multiple datasets by dataset group * update config file sanitizer * prune redundant validation * add comments * update type annotation * rename json_file_name to metadata_file * ignore when image dir is invalid * fix tag shuffle and dropout * ignore duplicated subset * add method to check latent cachability * fix format * fix bug * update caption dropout default values * update annotation * fix bug * add option to enable bucket shuffle across dataset * update blueprint generate function * use blueprint generator for dataset initialization * delete duplicated function * update config readme * delete debug print * print dataset and subset info as info * enable bucket_shuffle_across_dataset option * update config readme for clarification * compensate quotes for string option example * fix bug of bad usage of join * conserve trained metadata backward compatibility * enable shuffle in data loader by default * delete resolved TODO * add comment for image data handling * fix reference bug * fix undefined variable bug * prevent raise overwriting * assert image_dir and metadata_file validity * add debug message for ignoring subset * fix inconsistent import statement * loosen too strict validation on float value * sanitize argument parser separately * make image_dir optional for fine tuning dataset * fix import * fix trailing characters in print * parse flexible dataset config deterministically * use relative import * print supplementary message for parsing error * add note about different methods * add note of benefit of separate dataset * add error example * add note for english readme plan --------- Co-authored-by: Kohya S <52813779+kohya-ss@users.noreply.github.com>
27 lines
477 B
Plaintext
27 lines
477 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
|
|
toml==0.10.2
|
|
voluptuous==0.13.1
|
|
# 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
|
|
# for kohya_ss library
|
|
.
|