Skip to content

Version 0.6.2

Compare
Choose a tag to compare
@kohya-ss kohya-ss released this 24 Apr 14:19
· 866 commits to main since this release
25c8279

Updated at 23 Apr. 2023, 2023/4/23:

  • Fixed to log to TensorBoard when --logging_dir is specified and --log_with is not specified.
  • --logging_dirを指定し--log_withを指定しない場合に、以前と同様にTensorBoardへログ出力するよう修正しました。

Updated at 22 Apr. 2023, 2023/4/22:

  • Added support for logging to wandb. Please refer to PR #428. Thank you p1atdev!

    • wandb installation is required. Please install it with pip install wandb. Login to wandb with wandb login command, or set --wandb_api_key option for automatic login.
    • Please let me know if you find any bugs as the test is not complete.
  • You can automatically login to wandb by setting the --wandb_api_key option. Please be careful with the handling of API Key. PR #435 Thank you Linaqruf!

  • Improved the behavior of --debug_dataset on non-Windows environments. PR #429 Thank you tsukimiya!

  • Fixed --face_crop_aug option not working in Fine tuning method.

  • Prepared code to use any upscaler in gen_img_diffusers.py.

  • wandbへのロギングをサポートしました。詳細は PR #428をご覧ください。p1atdev氏に感謝します。

    • wandb のインストールが別途必要です。pip install wandb でインストールしてください。また wandb login でログインしてください(学習スクリプト内でログインする場合は --wandb_api_key オプションを設定してください)。
    • テスト未了のため不具合等ありましたらご連絡ください。
  • wandbへのロギング時に --wandb_api_key オプションを設定することで自動ログインできます。API Keyの扱いにご注意ください。 PR #435 Linaqruf氏に感謝します。

  • Windows以外の環境での--debug_dataset の動作を改善しました。PR #429 tsukimiya氏に感謝します。

  • --face_crop_augオプションがFine tuning方式で動作しなかったのを修正しました。

  • gen_img_diffusers.pyに任意のupscalerを利用するためのコード準備を行いました。

Updated at 19 Apr. 2023, 2023/4/19:

  • Fixed lora_interrogator.py not working. Please refer to PR #392 for details. Thank you A2va and heyalexchoi!
  • Fixed the handling of tags containing _ in tag_images_by_wd14_tagger.py.
  • lora_interrogator.pyが動作しなくなっていたのを修正しました。詳細は PR #392 をご参照ください。A2va氏およびheyalexchoi氏に感謝します。
  • tag_images_by_wd14_tagger.py_を含むタグの取り扱いを修正しました。

Updated at 17 Apr. 2023, 2023/4/17:

  • Added the --recursive option to each script in the finetune folder to process folders recursively. Please refer to PR #400 for details. Thanks to Linaqruf!
  • finetuneフォルダ内の各スクリプトに再起的にフォルダを処理するオプション--recursiveを追加しました。詳細は PR #400 を参照してください。Linaqruf 氏に感謝します。

Updated at 14 Apr. 2023, 2023/4/14:

  • Fixed a bug that caused an error when loading DyLoRA with the --network_weight option in train_network.py.
  • train_network.pyで、DyLoRAを--network_weightオプションで読み込むとエラーになる不具合を修正しました。

Updated at 13 Apr. 2023, 2023/4/13:

  • Added support for DyLoRA in train_network.py. Please refer to here for details (currently only in Japanese).

  • Added support for caching latents to disk in each training script. Please specify both --cache_latents and --cache_latents_to_disk options.

    • The files are saved in the same folder as the images with the extension .npz. If you specify the --flip_aug option, the files with _flip.npz will also be saved.
    • Multi-GPU training has not been tested.
    • This feature is not tested with all combinations of datasets and training scripts, so there may be bugs.
  • Added workaround for an error that occurs when training with fp16 or bf16 in fine_tune.py.

  • train_network.pyでDyLoRAをサポートしました。詳細はこちらをご覧ください。

  • 各学習スクリプトでlatentのディスクへのキャッシュをサポートしました。--cache_latentsオプションに 加えて--cache_latents_to_diskオプションを指定してください。

    • 画像と同じフォルダに、拡張子 .npz で保存されます。--flip_augオプションを指定した場合、_flip.npzが付いたファイルにも保存されます。
    • マルチGPUでの学習は未テストです。
    • すべてのDataset、学習スクリプトの組み合わせでテストしたわけではないため、不具合があるかもしれません。
  • fine_tune.pyで、fp16およびbf16の学習時にエラーが出る不具合に対して対策を行いました。