Commit
•
52df825
1
Parent(s):
6d6b652
Update config.json (#1)
Browse files- Update config.json (fa0185f8fab5714f5cfc9142b6755cd7706e20fb)
Co-authored-by: Yaowei Zheng <hiyouga@users.noreply.huggingface.co>
- config.json +2 -3
config.json
CHANGED
@@ -1,5 +1,4 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "saves/paligemma-3b-chat-v0.2",
|
3 |
"architectures": [
|
4 |
"PaliGemmaForConditionalGeneration"
|
5 |
],
|
@@ -19,7 +18,7 @@
|
|
19 |
"num_hidden_layers": 18,
|
20 |
"num_image_tokens": 1024,
|
21 |
"num_key_value_heads": 1,
|
22 |
-
"torch_dtype": "
|
23 |
"vocab_size": 257216
|
24 |
},
|
25 |
"torch_dtype": "bfloat16",
|
@@ -39,4 +38,4 @@
|
|
39 |
"vision_use_head": false
|
40 |
},
|
41 |
"vocab_size": 257216
|
42 |
-
}
|
|
|
1 |
{
|
|
|
2 |
"architectures": [
|
3 |
"PaliGemmaForConditionalGeneration"
|
4 |
],
|
|
|
18 |
"num_hidden_layers": 18,
|
19 |
"num_image_tokens": 1024,
|
20 |
"num_key_value_heads": 1,
|
21 |
+
"torch_dtype": "bfloat16",
|
22 |
"vocab_size": 257216
|
23 |
},
|
24 |
"torch_dtype": "bfloat16",
|
|
|
38 |
"vision_use_head": false
|
39 |
},
|
40 |
"vocab_size": 257216
|
41 |
+
}
|