QHRQQQ commited on
Commit
ffa77d3
·
verified ·
1 Parent(s): b046703

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

Browse files
flux/best_predictor_stage2_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
+ }