SDXL 用 LoRA 訓練《補足》WD 1.4 MOAT Tagger V2 (ブログ)

作成 : Masashi Okumura (@ClassCat)
作成日時 : 10/22/2023

* サンプルコードの動作確認はしておりますが、動作環境によりコードの追加変更が必要な場合はあるかもしれません。
* ご自由にリンクを張って頂いてかまいませんが、sales-info@classcat.com までご一報いただけると嬉しいです。

クラスキャット人工知能研究開発支援サービス

◆ クラスキャットは人工知能・テレワークに関する各種サービスを提供しています。お気軽にご相談ください :

人工知能研究開発支援
1. 人工知能研修サービス(経営者層向けオンサイト研修)
2. テクニカルコンサルティングサービス
3. 実証実験(プロトタイプ構築)
4. アプリケーションへの実装
人工知能研修サービス
PoC(概念実証)を失敗させないための支援

◆ 人工知能とビジネスをテーマに WEB セミナーを定期的に開催しています。スケジュール。

◆ お問合せ : 本件に関するお問い合わせ先は下記までお願いいたします。

SDXL 用 LoRA 訓練《補足》WD 1.4 MOAT Tagger V2

前回の記事 SDXL 用 LoRA 訓練 (on Google Colab) の補足として、WD 1.4 MOAT Tagger V2 によるタグ付け、キャプションファイルの生成について説明します。

WD 1.4 MOAT Tagger V2 は Automatic 1111 から利用可能です 🤓

前の記事に従って Automatic 1111 を起動します :

そして “Extensions” タブ、次いで “Available” タブをクリックすると以下のような画面に移行します :

ここでオレンジ色の “Load from:” ボタンをクリックすると、以下のように利用可能な拡張機能のリストが表示されます :

拡張機能のリストの中に “WD 1.4 Tagger tab, training” という項目がありますので :

右端の “install” ボタンをクリックしてインストールします。すると、以下のようなメッセージが出て、インストールされたタブを利用するには再起動が必要であると促されますので、再起動します :

Installed into /content/stable-diffusion-webui/extensions/stable-diffusion-webui-wd14-tagger. Use Installed tab to restart.

Stable Diffusion WebUI を再起動すると、以下のように “Tagger” タブが追加されていますのでクリックします :

すると次のような画面に移行しますが :

LoRA 用の訓練画像データセットに対してタグ付けするには “Batch from directory” タブをクリックします :

ここで WD 1.4 Tagger が利用可能になります。最小限必要な設定は :

Input directory (入力ディレクトリ) : 例えば、”/content/LoRA/train_data” です。
Output directory (出力ディレクトリ) : 入力ディレクトリと同じで問題ありません。
interrogator (tagger) : デフォルトは “WD14 moat tagger V2” です。
Additional tags (追加タグ) : 通常は、インスタンストークンやクラストークンを追加します。
weight threshold (重み閾値) : デフォルトは 0.35 ですが、一般にキャラクターの場合は高めに設定し、スタイルの場合は低めに設定します。

interrogator は以下が選択可能です :

設定が完了したら、最後にオレンジ色の “Interrogate” ボタンをクリックします。

以上