Update test_configs/mistral_test_config.yaml

test_configs/mistral_test_config.yaml

@@ -2,19 +2,19 @@ model:
   arch: mini_gpt4_llama_v2
   freeze_vit: True
   freeze_qformer: True
-  max_txt_len: 512
+  max_txt_len: 384
   low_resource: True
   image_size: 224
   end_sym: "</s>"
   llama_model: "mistralai/Mistral-7B-Instruct-v0.2"
-  ckpt: "checkpoints/video_mistral_all_checkpoint_last.pth"
+  ckpt: "checkpoints/video_mistral_checkpoint_last.pth"
   use_grad_checkpoint: True
   chat_template: True
   lora_r: 64
   lora_alpha: 16
-  length: 50
+  length: 32
   use_grad_checkpoint_llm: True
-  max_context_len: 7200
+  max_context_len: 4096
   architectures: [
     "MiniGPT4_Video"
   ]
@@ -22,9 +22,9 @@ model:
   drop_path_rate: 0 
   img_size: 224
   model_type: "minigpt4_video"
-  num_query_token: 32
+  num_query_token: 24
   prompt: ""
-  torch_dtype: "float32"
+  torch_dtype: "float16"
   transformers_version: "4.42.3"
   vit_precision: "fp16"
   vit_model: "eva_clip_g"