File size: 1,365 Bytes
7c65031 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | {
"format": "netra-bundle",
"format_version": 1,
"name": "netra-nmt",
"version": "1.0.0",
"task": "translation",
"languages": [
"en",
"km"
],
"graphs": {
"encoder": "encoder.onnx",
"decoder_step": "decoder_step.onnx"
},
"components": {
"preprocess": {
"type": "nmt-source",
"lang_markers": {
"en": "<2en>",
"km": "<2km>"
},
"max_len": 256,
"en_source": "nfc-collapse-whitespace-lowercase",
"km_source": "khmercut-segment"
},
"decoder": {
"type": "kv-cache",
"layers": 2,
"heads": 8,
"head_dim": 64,
"vocab_size": 32000,
"pad_id": 0,
"bos_id": 2,
"eos_id": 3
},
"tokenizer": {
"type": "sentencepiece",
"model": "spm_32k.model"
},
"postprocess": [
{
"type": "km-desegment"
}
]
},
"metadata": {
"source_checkpoint": "export_rope",
"architecture": "encoder-decoder, 12 enc / 2 dec layers, d_model 512, RoPE, RMSNorm, SwiGLU, tied embeddings",
"license": "MIT"
},
"files": {
"decoder_step.onnx": "23ca5422992f1cc388a1b58109bf1b68cc6352b35ef2e2f7c60c5c9cd5538a2d",
"encoder.onnx": "7e04885e0c170a2f945de8482c3b83da67fe44716a698875c151caf339b37eae",
"spm_32k.model": "213884dadb76370b50098e0b5cfeacad3d9813aa415657c7abdbb3bddde61ab7"
}
}
|