mirror of
https://www.modelscope.cn/OmniGen2/OmniGen2.git
synced 2026-04-02 18:12:55 +08:00
30 lines
552 B
JSON
30 lines
552 B
JSON
{
|
|
"_class_name": "OmniGen2Transformer2DModel",
|
|
"_diffusers_version": "0.33.1",
|
|
"axes_dim_rope": [
|
|
40,
|
|
40,
|
|
40
|
|
],
|
|
"axes_lens": [
|
|
1024,
|
|
1664,
|
|
1664
|
|
],
|
|
"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,
|
|
"use_fused_rms_norm": true,
|
|
"use_fused_swiglu": true
|
|
}
|