Update README.md
Browse files
README.md
CHANGED
|
@@ -162,8 +162,9 @@ display(output)
|
|
| 162 |
|
| 163 |
## Training:
|
| 164 |
|
| 165 |
-
|
| 166 |
### Stage C
|
|
|
|
| 167 |
**GPU used**: 7x Nvidia H100 80GB SXM5
|
| 168 |
| parameter | value |
|
| 169 |
|---|---|
|
|
@@ -178,8 +179,9 @@ Base model: Disty0/sotediffusion-wuerstchen3
|
|
| 178 |
| **images** | 6M * 2 captions per image |
|
| 179 |
| **epochs** | 1 |
|
| 180 |
|
| 181 |
-
|
| 182 |
### Stage B
|
|
|
|
| 183 |
**GPU used**: 1x Nvidia H100 80GB SXM5
|
| 184 |
| parameter | value |
|
| 185 |
|---|---|
|
|
|
|
| 162 |
|
| 163 |
## Training:
|
| 164 |
|
| 165 |
+
|
| 166 |
### Stage C
|
| 167 |
+
**Base model**: Disty0/sotediffusion-wuerstchen3
|
| 168 |
**GPU used**: 7x Nvidia H100 80GB SXM5
|
| 169 |
| parameter | value |
|
| 170 |
|---|---|
|
|
|
|
| 179 |
| **images** | 6M * 2 captions per image |
|
| 180 |
| **epochs** | 1 |
|
| 181 |
|
| 182 |
+
|
| 183 |
### Stage B
|
| 184 |
+
**Base model**: Disty0/sotediffusion-wuerstchen3-decoder
|
| 185 |
**GPU used**: 1x Nvidia H100 80GB SXM5
|
| 186 |
| parameter | value |
|
| 187 |
|---|---|
|