Update dit/config.json
Browse files- dit/config.json +1 -1
dit/config.json
CHANGED
|
@@ -40,7 +40,7 @@
|
|
| 40 |
"mmdit_sr",
|
| 41 |
"mmdit_sr"
|
| 42 |
],
|
| 43 |
-
"dtype": "
|
| 44 |
"emb_dim": 18432,
|
| 45 |
"expand_ratio": 4,
|
| 46 |
"head_dim": 128,
|
|
|
|
| 40 |
"mmdit_sr",
|
| 41 |
"mmdit_sr"
|
| 42 |
],
|
| 43 |
+
"dtype": "bfloat16",
|
| 44 |
"emb_dim": 18432,
|
| 45 |
"expand_ratio": 4,
|
| 46 |
"head_dim": 128,
|