Push model using huggingface_hub.
Browse files- 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,
|