| { | |
| "_name_or_path": "./", | |
| "architectures": [ | |
| "InfiMMHDModel" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration_infimm_hd.InfiMMHDConfig", | |
| "AutoModelForCausalLM": "modeling_infimm_hd.InfiMMHDModel" | |
| }, | |
| "model_type": "infimm-hd", | |
| "seq_length": 4096, | |
| "tokenizer_type": "LlamaTokenizer", | |
| "torch_dtype": "bfloat16", | |
| "transformers_version": "4.35.2", | |
| "use_cache": true, | |
| "use_flash_attn": false, | |
| "cross_attn_every_n_layers": 4, | |
| "use_grad_checkpoint": false, | |
| "freeze_llm": true, | |
| "image_token_id": 32001, | |
| "eoc_token_id": 32000, | |
| "visual": { | |
| "image_size": 448, | |
| "layers": 64, | |
| "width": 1792, | |
| "head_width": 112, | |
| "patch_size": 14, | |
| "mlp_ratio": 8.571428571428571, | |
| "eva_model_name": "eva-clip-4b-14-x", | |
| "drop_path_rate": 0.0, | |
| "xattn": false, | |
| "postnorm": true, | |
| "fusedLN": false, | |
| "embed_dim": 1024 | |
| }, | |
| "language": { | |
| "_name_or_path": "lmsys/vicuna-13b-v1.5", | |
| "architectures": [ | |
| "LlamaForCausalLM" | |
| ], | |
| "bos_token_id": 1, | |
| "eos_token_id": 2, | |
| "hidden_act": "silu", | |
| "hidden_size": 5120, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 13824, | |
| "max_position_embeddings": 4096, | |
| "model_type": "llama", | |
| "num_attention_heads": 40, | |
| "num_hidden_layers": 40, | |
| "num_key_value_heads": 40, | |
| "pad_token_id": 0, | |
| "pretraining_tp": 1, | |
| "rms_norm_eps": 1e-05, | |
| "rope_scaling": null, | |
| "tie_word_embeddings": false, | |
| "torch_dtype": "bfloat16", | |
| "transformers_version": "4.32.0.dev0", | |
| "use_cache": true, | |
| "vocab_size": 32002 | |
| } | |
| } |