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 |

|
| 46 |

|
| 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 |

|
| 56 |

|
| 57 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
#### Credits
|
| 59 |
|
| 60 |
- Sta, AI Novelist Dev (https://ai-novel.com/) @ Bit192, Inc.
|