Fix VoxCPM2 LiteRT semantic roundtrip
Browse files- config.json +12 -0
- voxcpm2-text-prefill.tflite +2 -2
config.json
CHANGED
|
@@ -229,6 +229,12 @@
|
|
| 229 |
"export": {
|
| 230 |
"source_repo": "openbmb/VoxCPM2",
|
| 231 |
"voxcpm_src_required": true,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 232 |
"token_step_cache_contract": "[2, layers, batch, kv_heads, max_cache_length, head_dim]",
|
| 233 |
"token_step_status": "experimental until litert_torch cache lowering is verified"
|
| 234 |
},
|
|
@@ -271,6 +277,12 @@
|
|
| 271 |
512
|
| 272 |
],
|
| 273 |
"name": "serving_default_args_3"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 274 |
}
|
| 275 |
},
|
| 276 |
"token_step": {
|
|
|
|
| 229 |
"export": {
|
| 230 |
"source_repo": "openbmb/VoxCPM2",
|
| 231 |
"voxcpm_src_required": true,
|
| 232 |
+
"graph_variants": {
|
| 233 |
+
"text_prefill": "fp32",
|
| 234 |
+
"token_step": "int8",
|
| 235 |
+
"audio_encoder": "fp32",
|
| 236 |
+
"audio_decoder": "fp32"
|
| 237 |
+
},
|
| 238 |
"token_step_cache_contract": "[2, layers, batch, kv_heads, max_cache_length, head_dim]",
|
| 239 |
"token_step_status": "experimental until litert_torch cache lowering is verified"
|
| 240 |
},
|
|
|
|
| 277 |
512
|
| 278 |
],
|
| 279 |
"name": "serving_default_args_3"
|
| 280 |
+
},
|
| 281 |
+
"context_length": {
|
| 282 |
+
"dtype": "int64",
|
| 283 |
+
"rank": 0,
|
| 284 |
+
"shape": [],
|
| 285 |
+
"name": "serving_default_args_4"
|
| 286 |
}
|
| 287 |
},
|
| 288 |
"token_step": {
|
voxcpm2-text-prefill.tflite
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bb63a91e00816b5ca592d14003febb6715ca7d78d789b9298b0f56d06f692a90
|
| 3 |
+
size 8280692912
|