Terra / world_model_v2 /config.json
koukyo1994's picture
update v2 weights
950a943 verified
{
"action_dim": 3,
"activation_function": "relu",
"architectures": [
"LlamaActionV2ForCausalLM"
],
"auto_map": {
"AutoConfig": "turing-motors/Terra--configuration_llama_action.LlamaActionConfig",
"AutoModel": "turing-motors/Terra--modeling_llama_action.LlamaActionV2ForCausalLM"
},
"attention_bias": false,
"attention_dropout": 0.0,
"bos_token_id": 262144,
"do_layer_norm_before": true,
"dropout": 0.1,
"enable_bias": true,
"eos_token_id": 262144,
"hidden_act": "silu",
"hidden_size": 2048,
"init_std": 0.02,
"initializer_range": 0.02,
"intermediate_size": 5632,
"layer_norm_elementwise_affine": true,
"layerdrop": 0.0,
"llm_model_type": "llama",
"max_position_embeddings": 14550,
"model_type": "llama_action",
"num_action_embeddings": 6,
"num_action_tokens": 6,
"num_attention_heads": 32,
"num_hidden_layers": 22,
"num_image_patches": 576,
"num_key_value_heads": 4,
"num_spatio_embeddings": 582,
"num_temporal_embeddings": 25,
"pad_token_id": 262144,
"pretraining_tp": 1,
"rms_norm_eps": 1e-06,
"rope_scaling": null,
"rope_theta": 10000.0,
"tie_word_embeddings": false,
"torch_dtype": "bfloat16",
"transformers_version": "4.39.3",
"use_cache": true,
"use_text": false,
"vocab_size": 262145,
"word_embedding_proj_dim": 2048
}