File size: 897 Bytes
ee88881 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | {
"format": "trinity-release-v1",
"arch": {
"latent_dim": 3,
"feature_dim": 19,
"graph_pe_dim": 2,
"hidden": 768,
"depth": 12,
"heads": 12,
"head_dim": 64,
"norm": "rmsnorm",
"mlp": "swiglu",
"mlp_ratio": 4.0,
"attn": "sdpa_graph",
"qk_norm": true,
"graph_bias": true,
"graph_mix": {
"name": "mp",
"normalize": "row"
},
"time_cond": "token",
"output_kind": "x0",
"post_norm": false,
"time_scale": 1000.0,
"grad_ckpt": false
},
"backbone": null,
"latent_param": "s_only",
"framing": "rectified_flow",
"graph_pe": {
"name": "spectral_draw",
"k": 2,
"normalize": "log1p"
},
"cond_options": {
"pin_pull_normalize": true,
"graph_bias_normalize": true,
"mib_anchor": "group_mean"
},
"sampler": {
"name": "euler",
"num_steps": 16
},
"name": "flagship"
}
|