QHRQQQ commited on
Commit
cbe0b00
·
verified ·
1 Parent(s): 2a2d7c3

Add per-stage predictor config for FLUX stage0 (required by from_pretrained)

Browse files
flux/best_predictor_stage0_config.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dim": 3072,
3
+ "num_blocks": 1,
4
+ "hidden_dim": 128,
5
+ "split_dims": [
6
+ 2304,
7
+ 384,
8
+ 384
9
+ ],
10
+ "dropout": 0.0
11
+ }