Vision-CAIR commited on
Commit
0c189dc
·
verified ·
1 Parent(s): e4a30a9

Push model using huggingface_hub.

Browse files
Files changed (1) hide show
  1. config.json +8 -0
config.json CHANGED
@@ -1,5 +1,12 @@
1
  {
2
  "arch": "mini_gpt4_llama_v2",
 
 
 
 
 
 
 
3
  "chat_template": true,
4
  "ckpt": "checkpoints/video_mistral_all_checkpoint_last.pth",
5
  "device": "cuda",
@@ -27,6 +34,7 @@
27
  "prompt_path": "",
28
  "remove_template": false,
29
  "token_pooling": true,
 
30
  "transformers_version": "4.42.3",
31
  "use_grad_checkpoint": true,
32
  "use_grad_checkpoint_llm": true,
 
1
  {
2
  "arch": "mini_gpt4_llama_v2",
3
+ "architectures": [
4
+ "MiniGPT4_Video"
5
+ ],
6
+ "auto_map": {
7
+ "AutoConfig": "mini_gpt4_llama_v2.minigpt4_video_config",
8
+ "AutoModel": "mini_gpt4_llama_v2.MiniGPT4_Video"
9
+ },
10
  "chat_template": true,
11
  "ckpt": "checkpoints/video_mistral_all_checkpoint_last.pth",
12
  "device": "cuda",
 
34
  "prompt_path": "",
35
  "remove_template": false,
36
  "token_pooling": true,
37
+ "torch_dtype": "float32",
38
  "transformers_version": "4.42.3",
39
  "use_grad_checkpoint": true,
40
  "use_grad_checkpoint_llm": true,