naclbit commited on
Commit
af3bbd5
1 Parent(s): 9c1a7cc

Added training info

Browse files
Files changed (1) hide show
  1. README.md +10 -4
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.