File size: 1,479 Bytes
52699fd 088eb9a 52699fd dc963f2 52699fd dc963f2 52699fd a9be2ba 337cbfd a9be2ba 52699fd |
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 |
{
"_name_or_path": "./",
"architectures": [
"InternVideo2_cls"
],
"auto_map": {
"AutoConfig": "model_config.VideoChat2Config",
"AutoModel": "modeling_videochat2_cls.InternVideo2_cls"
},
"model_cls": "InternVideo2_VideoChat2",
"model_config": {
"bridge": {
"extra_num_query_token": 64,
"name": "qformer",
"num_query_token": 32,
"qformer_attention_probs_dropout_prob": 0.1,
"qformer_drop_path_rate": 0.2,
"qformer_hidden_dropout_prob": 0.1
},
"freeze_bridge": false,
"freeze_llm": false,
"freeze_vision_encoder": false,
"llm": {
"lora_alpha": 32,
"lora_dropout": 0.1,
"lora_r": 16,
"name": "mistral_7b",
"num_labels": 512,
"pretrained_llm_path": "mistralai/Mistral-7B-Instruct-v0.3",
"use_lora": true
},
"pretrained_paths": {},
"use_flash_attention": true,
"vision_encoder": {
"checkpoint_num": 48,
"d_model": 1408,
"encoder_embed_dim": 1408,
"img_size": 224,
"name": "internvideo2-1B",
"num_frames": 8,
"origin_num_frames": 4,
"patch_size": 14,
"pretrained": null,
"sep_image_video_pos_embed": true,
"tubelet_size": 1,
"use_checkpoint": true,
"vit_add_ln": true,
"x_vis_only": true,
"x_vis_return_idx": -2
}
},
"model_type": "InternVideo2_VideoChat2",
"torch_dtype": "float32",
"transformers_version": "4.46.1",
"use_cache": true
}
|