Set default dtypes
Browse files- config.json +8 -0
config.json
CHANGED
|
@@ -55,6 +55,14 @@
|
|
| 55 |
"kv_cache_dtype": {
|
| 56 |
"fp16": "float16",
|
| 57 |
"q4f16": "float16"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 58 |
}
|
| 59 |
},
|
| 60 |
"vision_config": {
|
|
|
|
| 55 |
"kv_cache_dtype": {
|
| 56 |
"fp16": "float16",
|
| 57 |
"q4f16": "float16"
|
| 58 |
+
},
|
| 59 |
+
"dtype": {
|
| 60 |
+
"prepare_inputs_embeds": "fp32",
|
| 61 |
+
"language_model": "q4",
|
| 62 |
+
"lm_head": "fp32",
|
| 63 |
+
"gen_head": "fp32",
|
| 64 |
+
"gen_img_embeds": "fp32",
|
| 65 |
+
"image_decode": "fp32",
|
| 66 |
}
|
| 67 |
},
|
| 68 |
"vision_config": {
|