File size: 1,072 Bytes
3807da3 | 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 | {
"model_name": "fireredpunc",
"framework": "pytorch",
"target_hardware": "AX650",
"npu_mode": "NPU1",
"compiler": "Pulsar2 7.0 via Magnetar COMPILE",
"quant": {
"activation": "U16",
"weight": "S8",
"gather_embedding": "FP32",
"smooth_quant": true
},
"seq_len": 128,
"inputs": [
{
"name": "input_ids",
"shape": [1, 128],
"dtype": "int32",
"layout": "NCHW"
},
{
"name": "attention_mask",
"shape": [1, 128],
"dtype": "float32",
"layout": "NCHW"
}
],
"outputs": [
{
"name": "logits",
"shape": [1, 127, 5],
"dtype": "float32"
}
],
"layout": "[CLS] + tokens, no [SEP]; logits row i belongs to token i",
"labels": [" ", "，", "。", "？", "！"],
"files": {
"axmodel": "punc.axmodel",
"tokenizer_dir": "tokenizer/",
"vocab": "vocab.txt",
"out_dict": "out_dict"
},
"source": "FireRedTeam/FireRedPunc (Apache-2.0), chinese-lert-base",
"notes": "Input must be unpunctuated; strip ，。？！、,.?! before inference."
}
|