File size: 1,492 Bytes
1fe2608 | 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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | {
"_class_name": "Cosmos3OmniTransformer",
"_diffusers_version": "0.39.0.dev0",
"_name_or_path": "nvidia/Cosmos3-Super-Text2Image",
"action_dim": 32,
"action_gen": false,
"attention_bias": false,
"attention_dropout": 0.0,
"base_fps": 24,
"dtype": "bfloat16",
"enable_fps_modulation": true,
"freeze_und": false,
"head_dim": 128,
"hidden_act": "silu",
"hidden_size": 5120,
"initializer_range": 0.02,
"intermediate_size": 25600,
"joint_attn_implementation": "two_way",
"latent_channel": 48,
"latent_patch_size": 2,
"max_action_dim": 32,
"max_position_embeddings": 262144,
"model_type": "qwen3_vl_text",
"num_attention_heads": 64,
"num_embodiment_domains": 32,
"num_hidden_layers": 64,
"num_key_value_heads": 8,
"patch_latent_dim": 192,
"position_embedding_type": "unified_3d_mrope",
"qk_norm": false,
"qk_norm_for_diffusion": true,
"qk_norm_for_text": true,
"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,
"temporal_compression_factor_sound": 1,
"timestep_scale": 0.001,
"unified_3d_mrope_reset_spatial_ids": true,
"unified_3d_mrope_temporal_modality_margin": 15000,
"use_cache": true,
"use_moe": true,
"video_temporal_causal": false,
"vocab_size": 151936
}
|