|
--- |
|
license: openrail++ |
|
thumbnail: >- |
|
https://huggingface.co/inu-ai/niji-diffusion-xl-base-1.0/images/thumbnail.png |
|
tags: |
|
- text-to-image |
|
- stable-diffusion |
|
- safetensors |
|
- diffusers |
|
inference: true |
|
widget: |
|
- text: >- |
|
best quality, high quality, absurdres, 1girl |
|
datasets: |
|
- p1atdev/niji-v5 |
|
library_name: diffusers |
|
--- |
|
# 更新履歴 |
|
|
|
- 2023年8月14日 |
|
|
|
さらにnijijourneyの画像1000枚ほどを目で見てアニメやイラスト風で良いもの選んで学習させました。 |
|
|
|
以下に何をやったか記録しますが、何が効いているのかわかりません。 |
|
|
|
以下にハイパーパラメータを記載しますが、その後[v11](https://civitai.com/models/120452?modelVersionId=135754)と[sd-webui-supermerger](https://github.com/hako-mikan/sd-webui-supermerger)階層マージをしているため一度で仕上がりませんでした。（block_lrに近い比率で絵が良さそうなモデル） |
|
最後に[blur](https://civitai.com/models/120624)を-0.05ほど、[LECO](https://github.com/p1atdev/LECO)で作ったanimeを1でLoRAマージ仕上げました。 |
|
|
|
| ハイパーパラメータ | 値 | |
|
|-----------------------|-------------------------------------| |
|
| GPU | RTX3090 24GB | |
|
| optimizer_type | PagedLion8bit | |
|
| optimizer_args | weight_decay=0.01, betas=.9,.999 | |
|
| block_lr | 0,1e-08,1e-08,1e-08,1e-08,1e-07,1e-07,1e-07,1e-06,1e-06,1e-05,1e-05,1e-05,1e-06,1e-06,1e-07,1e-07,1e-07,1e-08,1e-08,1e-08,1e-08,0 | |
|
| lr_scheduler | cosine | |
|
| lr_warmup_steps | 100 | |
|
| gradient_checkpointing| | |
|
| mixed_precision | bf16 | |
|
| full_bf16 | | |
|
| max_token_length | 225 | |
|
| min_snr_gamma | 5 | |
|
| noise_offset | 0.0357 | |
|
| max_train_epochs | 3 | |
|
| batch_size | 12 | |
|
| enable_bucket | true | |
|
| resolution | [1024,1024] | |
|
|
|
- 2023年8月11日 |
|
前回のnijijourneyの画像を混ぜて12000枚で学習しました。 |
|
|
|
オプティマイザはLion(4e-06, cosine, weight_decay=0.015, betas=.9,.999)でした。 |
|
|
|
|
|
- 2023年8月7日 |
|
|
|
nijijourneyの画像4500枚ぐらいでフルファインチューニングしました。fp16で壊れないVAEに取り替えました。 |
|
|
|
学習率1e-07が低すぎたようで絵があまり変化しませんでした。次回は学習率を上げようと思います。 |
|
|
|
- 2023年8月1日 |
|
|
|
nijijourneyの画像100枚ぐらいでLoRAファインチューニングなどをしました。 |
|
|
|
# モデルの説明 |
|
アニメ調に寄せた「[SDXL(stable-diffusion-xl-base-1.0)](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0)」モデルです。内容は「[niji-v5](https://huggingface.co/datasets/p1atdev/niji-v5)」データセットでLoRAファインチューニングしてマージしたモデルです。 |
|
|
|
# 使用例 |
|
|
|
[niji-diffusion-xl-base-1.0.safetensors](https://huggingface.co/inu-ai/niji-diffusion-xl-base-1.0/resolve/main/niji-diffusion-xl-base-1.0.safetensors)と[stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)で以下のパラメータで画像生成します。 |
|
|
|
※合計13000枚ぐらい~100枚~しか学習していないのでPromptに複数書くとnijiの絵じゃなくなります。Negative promptに複数書くのは大丈夫っぽいです。 |
|
|
|
Prompt: |
|
``` |
|
1girl, town |
|
``` |
|
Negative prompt: |
|
``` |
|
worst quality, low quality, medium quality, deleted, lowres, comic, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, jpeg artifacts, signature, watermark, username, blurry |
|
``` |
|
|
|
PNG info: |
|
``` |
|
Steps: 28, Sampler: Euler a, CFG scale: 7, Seed: 1, Size: 1536x1024, Model hash: 791d0c791e, Model: sd_xl_niji_1.0, Clip skip: 2, ENSD: 31337, Token merging ratio: 0.5, Eta: 0.67, Version: v1.5.1 |
|
``` |
|
 |
|
|
|
Prompt: |
|
``` |
|
1girl |
|
``` |
|
 |
|
Prompt: |
|
``` |
|
1girl, tokyo |
|
``` |
|
 |
|
Prompt: |
|
``` |
|
1girl, steampunk |
|
``` |
|
 |
|
Prompt: |
|
``` |
|
1girl, fantasy |
|
``` |
|
 |
|
|
|
# モデルの作り方 |
|
1. 「[かんたん☆コピー機学習法（きっと初級編）](https://note.com/2vxpswa7/n/n2d04527bf0bc)」を参考に、「[ぼかし（blur）](https://civitai.com/models/120624)」のLoRA DreamBoothをして、そのLoRAモデルをSDXLモデルにマイナスでマージします。 |
|
2. 1で作ったモデルに、niji-v5から背景や髪が詳細に描かれている絵などを100枚選びLoRAファインチューニングして、そのLoRAモデルをSDXLモデルにマージします。 |
|
|
|
# 今後のモデルの改善 |
|
LoRAモデルとして配布できるようにしたいです。 |
|
というのは、512dim(rank)で学習したらLoRAモデルが3GBのファイルサイズになってしまったため、今回はSDXLモデルにマージしています。 |
|
|
|
# 感想 |
|
うまく調整できず何度もやり直して大変でした。あとで作り方を動画にしたいです。 |
|
|
|
# 謝辞 |
|
モデル、学習データや学習ツールを作成および配布してくださった方々に心から感謝申し上げます。 |
|
|
|
# ライブラリ |
|
|
|
- [sd-scripts](https://github.com/kohya-ss/sd-scripts/tree/sdxl) 4072f723c12822e2fa1b2e076cc1f90b8f4e30c9 |
|
- [bitsandbytes](https://github.com/jllllll/bitsandbytes-windows-webui) 0.39.1 |
|
- Pytorch 2.0.0+cu117 |
|
- xformers 0.0.19 |
|
|
|
# ライセンス |
|
- [CreativeML Open RAIL++-M License](https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/blob/main/LICENSE.md) |
