File size: 556 Bytes
e5e3587 eb214b7 e5e3587 |
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 |
{
"_class_name": "OmniGen2Transformer3DModel",
"_diffusers_version": "0.34.0",
"_name_or_path": "pretrained_models/ft_023/transformer",
"axes_dim_rope": [
40,
40,
40
],
"axes_lens": [
10000,
10000,
10000
],
"ffn_dim_multiplier": null,
"hidden_size": 2520,
"in_channels": 16,
"multiple_of": 256,
"norm_eps": 1e-05,
"num_attention_heads": 21,
"num_kv_heads": 7,
"num_layers": 32,
"num_refiner_layers": 2,
"out_channels": null,
"patch_size": 2,
"text_feat_dim": 2048,
"timestep_scale": 1000.0
}
|