File size: 1,945 Bytes
0237000 |
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 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
{
"_name_or_path": "xvla",
"model_type": "xvla",
"architectures": ["XVLA"],
"auto_map": {
"AutoConfig": "configuration_xvla.XVLAConfig",
"AutoModel": "modeling_xvla.XVLA"
},
"action_mode": "ee6d",
"use_proprio": true,
"num_actions": 30,
"hidden_size": 1024,
"depth": 24,
"num_heads": 16,
"mlp_ratio": 4.0,
"num_domains": 30,
"len_soft_prompts": 32,
"dim_time": 32,
"max_len_seq": 512,
"use_hetero_proj": false,
"soft_prompt_length": 32,
"florence_config": {
"model_type": "florence2",
"bos_token_id": 0,
"eos_token_id": 2,
"ignore_index": -100,
"pad_token_id": 1,
"projection_dim": 1024,
"text_config": {
"vocab_size": 51289,
"activation_dropout": 0.1,
"activation_function": "gelu",
"attention_dropout": 0.1,
"d_model": 1024,
"decoder_attention_heads": 16,
"decoder_layers": 12,
"encoder_attention_heads": 16,
"encoder_layers": 12,
"dropout": 0.1,
"max_position_embeddings": 4096,
"num_hidden_layers": 12,
"num_beams": 3
},
"vision_config": {
"model_type": "davit",
"drop_path_rate": 0.1,
"patch_size": [7, 3, 3, 3],
"patch_stride": [4, 2, 2, 2],
"patch_padding": [3, 1, 1, 1],
"patch_prenorm": [false, true, true, true],
"enable_checkpoint": false,
"dim_embed": [256, 512, 1024, 2048],
"num_heads": [8, 16, 32, 64],
"num_groups": [8, 16, 32, 64],
"depths": [1, 1, 9, 1],
"window_size": 12,
"projection_dim": 1024,
"visual_temporal_embedding": {
"type": "COSINE",
"max_temporal_embeddings": 100
},
"image_pos_embed": {
"type": "learned_abs_2d",
"max_pos_embeddings": 50
},
"image_feature_source": ["spatial_avg_pool", "temporal_avg_pool"]
},
"vocab_size": 51289,
"torch_dtype": "float16",
"is_encoder_decoder": true
}
}
|