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