benjamin-paine commited on
Commit
873b590
·
verified ·
1 Parent(s): b6397e4

Update dit/config.json

Browse files
Files changed (1) hide show
  1. dit/config.json +1 -1
dit/config.json CHANGED
@@ -40,7 +40,7 @@
40
  "mmdit_sr",
41
  "mmdit_sr"
42
  ],
43
- "dtype": "float32",
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,