furusu commited on
Commit
aaf2357
·
1 Parent(s): 7c9f81d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -1
README.md CHANGED
@@ -4,7 +4,37 @@ tags:
4
  - stable-diffusion
5
  - text-to-image
6
  ---
 
7
 
8
- This model is trained from [SDv2-1](https://huggingface.co/stabilityai/stable-diffusion-2-1) for 10 epochs on 130k images and then 4 epochs on 390k images.
 
9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
 
 
4
  - stable-diffusion
5
  - text-to-image
6
  ---
7
+ # th-diffusion
8
 
9
+ このモデルは、[SD2-1(768)](https://huggingface.co/stabilityai/stable-diffusion-2-1)からアニメスタイルの画像を学習させたものです。SDからアニメスタイルを自力で作ってみたかっただけです。
10
+ 学習方法は[WD1-4](https://huggingface.co/hakurei/waifu-diffusion-v1-4)とほとんど同じであり、データセットも学習ステップ数も負けてるので劣化版でしかないです。
11
 
12
+ このモデルの学習は三段階に分かれます。
13
+
14
+ 1. 13万枚の画像を10エポック学習:A100 80GBで20時間くらい
15
+ 2. 39万枚の画像をText endoderを含めて1エポック・含めずに追加で3エポック学習:RTX3090で80時間くらい
16
+ 4. 61万枚の画像を3エポック学習:RTX3090で100時間くらい
17
+
18
+ 学習設定:
19
+
20
+ + 解像度 768×768を基準にしたAspect ratio bucketing.半分くらいが縦896横640です。
21
+ + 学習率 5e-6のconstant
22
+ + batch size 1.は20 2.3.は16
23
+ + タグ [WD1-4tagger](https://huggingface.co/SmilingWolf/wd-v1-4-vit-tagger)を利用しました。
24
+
25
+
26
+ 2.3.は[WD1-4のタグ付け規則](https://gist.github.com/harubaru/313eec09026bb4090f4939d01f79a7e7)に従っています。
27
+ そのため機能するプロンプトやネガティブプロンプトはWD1-4と変わりません。
28
+
29
+ # 生成例
30
+ てきとう、ネガティブプロンプトは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
31
+
32
+ CFG_scaleは10、高めにしたほうがいいかも。
33
+
34
+ masterpiece,best quality,1girl,solo,sitting,blush,red eyes,blonde hair,twintails,hair ribbon,school uniform,blue sailor collar,blue skirt,black thighhighs
35
+ ![image](grid-0084.png)
36
+ masterpiece,best quality,absurdres,safe,1girl,solo,one eye closed, brown hair, side ponytail, maid, maid headdress, white thighhighs
37
+ ![image](grid-0082.png)
38
+ masterpiece,best quality,hakurei reimu,touhou,solo,blush,brown hair,red eyes,frills,navel , yellow ascot, detached sleeves,hair bow,hair tubes
39
+ ![image](grid-0087.png)
40