upload model folder to repo

This commit is contained in:
lxlxlxlxlxlx
2025-06-16 07:50:43 +00:00
parent 43f9976329
commit cd09780d2b
37 changed files with 307054 additions and 1 deletions

29
transformer/config.json Normal file
View File

@ -0,0 +1,29 @@
{
"_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
}