Added training info
Browse files
README.md
CHANGED
@@ -37,6 +37,16 @@ If you want to run **latent-diffusion**'s stock ddim img2img script with this mo
|
|
37 |
|
38 |
**latent-diffusion** のscriptsフォルダに入っているddim img2imgをこのモデルで動かす場合、use_emaはFalseにする必要があります。
|
39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
#### Examples
|
41 |
|
42 |
Each images were diffused using K. Crowson's k-lms (from k-diffusion repo) method for 50 steps.
|
@@ -45,10 +55,6 @@ Each images were diffused using K. Crowson's k-lms (from k-diffusion repo) metho
|
|
45 |
![examples](https://pbs.twimg.com/media/FbPO65cUIAAga8k?format=jpg&name=900x900)
|
46 |
![examples](https://pbs.twimg.com/media/FbPO_QuVsAAG6xE?format=png&name=900x900)
|
47 |
|
48 |
-
#### Hardware
|
49 |
-
|
50 |
-
- 8xNVIDIA A100 40GB
|
51 |
-
|
52 |
#### Credits
|
53 |
|
54 |
- Sta, AI Novelist Dev (https://ai-novel.com/) @ Bit192, Inc.
|
|
|
37 |
|
38 |
**latent-diffusion** のscriptsフォルダに入っているddim img2imgをこのモデルで動かす場合、use_emaはFalseにする必要があります。
|
39 |
|
40 |
+
#### Hardware
|
41 |
+
|
42 |
+
- 8xNVIDIA A100 40GB
|
43 |
+
|
44 |
+
#### Training Info
|
45 |
+
|
46 |
+
- Custom dataset loader with augmentations: XFlip, center crop and aspect-ratio locked scaling
|
47 |
+
- LR: 1.0e-5
|
48 |
+
- 10% dropouts
|
49 |
+
|
50 |
#### Examples
|
51 |
|
52 |
Each images were diffused using K. Crowson's k-lms (from k-diffusion repo) method for 50 steps.
|
|
|
55 |
![examples](https://pbs.twimg.com/media/FbPO65cUIAAga8k?format=jpg&name=900x900)
|
56 |
![examples](https://pbs.twimg.com/media/FbPO_QuVsAAG6xE?format=png&name=900x900)
|
57 |
|
|
|
|
|
|
|
|
|
58 |
#### Credits
|
59 |
|
60 |
- Sta, AI Novelist Dev (https://ai-novel.com/) @ Bit192, Inc.
|