| { | |
| "_name_or_path": "./checkpoints-pretrain/vstream-vicuna-7b-pretrain-weighted_kmeans1*8-25*4-25*1/checkpoint-3000", | |
| "architectures": [ | |
| "VStreamLlamaForCausalLM" | |
| ], | |
| "bos_token_id": 1, | |
| "compress_Turing_memory_size": 1, | |
| "compress_Turing_update_ratio": 0.2, | |
| "compress_long_memory_size": 4, | |
| "compress_size": 8, | |
| "compress_type": "mean", | |
| "eos_token_id": 2, | |
| "freeze_mm_mlp_adapter": false, | |
| "freeze_mm_vision_resampler": false, | |
| "hidden_act": "silu", | |
| "hidden_size": 4096, | |
| "image_aspect_ratio": "pad", | |
| "initializer_range": 0.02, | |
| "intermediate_size": 11008, | |
| "max_position_embeddings": 4096, | |
| "mm_hidden_size": 1024, | |
| "mm_projector_lr": null, | |
| "mm_projector_type": "mlp2x_gelu", | |
| "mm_resampler_type": null, | |
| "mm_use_4_vision_tokens": false, | |
| "mm_use_im_patch_token": false, | |
| "mm_use_im_start_end": false, | |
| "mm_vision_select_feature": "patch", | |
| "mm_vision_select_layer": -2, | |
| "mm_vision_tower": "openai/clip-vit-large-patch14", | |
| "model_type": "vstream", | |
| "num_attention_heads": 32, | |
| "num_hidden_layers": 32, | |
| "num_key_value_heads": 32, | |
| "pad_token_id": 0, | |
| "pretraining_tp": 1, | |
| "rms_norm_eps": 1e-05, | |
| "rope_scaling": null, | |
| "tie_word_embeddings": false, | |
| "tokenizer_model_max_length": 2048, | |
| "tokenizer_padding_side": "right", | |
| "torch_dtype": "bfloat16", | |
| "transformers_version": "4.31.0", | |
| "tune_mm_mlp_adapter": false, | |
| "tune_mm_vision_resampler": false, | |
| "unfreeze_mm_vision_tower": false, | |
| "use_cache": false, | |
| "use_mm_proj": true, | |
| "video_Turing_memory_length": 25, | |
| "video_current_memory_length": 1, | |
| "video_long_memory_length": 25, | |
| "video_max_frames": 26, | |
| "video_sample_type": "weighted_kmeans", | |
| "video_short_memory_length": 10, | |
| "vocab_size": 32000 | |
| } | |