File size: 952 Bytes
6319a87 | 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 | {
"_class_name": "Cosmos3OmniTransformer",
"_diffusers_version": "0.39.0",
"action_dim": 32,
"action_gen": false,
"attention_bias": false,
"attention_dropout": 0.0,
"base_fps": 24,
"enable_fps_modulation": true,
"head_dim": 128,
"hidden_size": 5120,
"intermediate_size": 25600,
"latent_channel": 48,
"latent_patch_size": 2,
"num_attention_heads": 64,
"num_embodiment_domains": 32,
"num_hidden_layers": 64,
"num_key_value_heads": 8,
"patch_latent_dim": 192,
"rms_norm_eps": 1e-06,
"rope_axes_dim": [
24,
20,
20
],
"rope_scaling": {
"mrope_interleaved": true,
"mrope_section": [
24,
20,
20
],
"rope_type": "default"
},
"rope_theta": 5000000,
"sound_dim": 64,
"sound_gen": true,
"sound_latent_fps": 25,
"timestep_scale": 0.001,
"unified_3d_mrope_reset_spatial_ids": true,
"unified_3d_mrope_temporal_modality_margin": 15000,
"vocab_size": 151936
}
|