raw float weights: model.safetensors + config.json, exported from the shipped model.onnx
Browse files- README.md +5 -0
- config.json +69 -0
- model.safetensors +3 -0
README.md
CHANGED
|
@@ -254,6 +254,11 @@ Flat layout, named per **runtime × GPU arch × OS** (single-profile — no leng
|
|
| 254 |
- **ONNX** — `model.onnx` FP32 dynamic-shape (the build source, and standalone `onnxruntime` /
|
| 255 |
`optimum` use), `model_static.onnx` (seq pinned to 128, the engine-build input), and
|
| 256 |
`model_int8qdt.onnx` (INT8 Q/DQ).
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 257 |
|
| 258 |
No MLX build: this model has no Apple-GPU consumer today.
|
| 259 |
|
|
|
|
| 254 |
- **ONNX** — `model.onnx` FP32 dynamic-shape (the build source, and standalone `onnxruntime` /
|
| 255 |
`optimum` use), `model_static.onnx` (seq pinned to 128, the engine-build input), and
|
| 256 |
`model_int8qdt.onnx` (INT8 Q/DQ).
|
| 257 |
+
- **Raw weights** — `model.safetensors` + `config.json`, the FP32 weights in the
|
| 258 |
+
`XLMRobertaForTokenClassification` layout with the 17 labels in the config:
|
| 259 |
+
`AutoModelForTokenClassification` loads them with no missing or unexpected keys, and its
|
| 260 |
+
per-token logits match the ONNX to 1e-5. Calibrate them with `temperature.json` exactly as
|
| 261 |
+
in the snippet above.
|
| 262 |
|
| 263 |
No MLX build: this model has no Apple-GPU consumer today.
|
| 264 |
|
config.json
ADDED
|
@@ -0,0 +1,69 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"XLMRobertaForTokenClassification"
|
| 4 |
+
],
|
| 5 |
+
"model_type": "xlm-roberta",
|
| 6 |
+
"attention_probs_dropout_prob": 0.1,
|
| 7 |
+
"bos_token_id": 0,
|
| 8 |
+
"eos_token_id": 2,
|
| 9 |
+
"hidden_act": "gelu",
|
| 10 |
+
"hidden_dropout_prob": 0.1,
|
| 11 |
+
"initializer_range": 0.02,
|
| 12 |
+
"num_attention_heads": 12,
|
| 13 |
+
"pad_token_id": 1,
|
| 14 |
+
"position_embedding_type": "absolute",
|
| 15 |
+
"vocab_size": 250002,
|
| 16 |
+
"hidden_size": 384,
|
| 17 |
+
"num_hidden_layers": 12,
|
| 18 |
+
"intermediate_size": 1536,
|
| 19 |
+
"max_position_embeddings": 514,
|
| 20 |
+
"type_vocab_size": 1,
|
| 21 |
+
"layer_norm_eps": 1e-05,
|
| 22 |
+
"id2label": {
|
| 23 |
+
"0": "O",
|
| 24 |
+
"1": "B-component",
|
| 25 |
+
"2": "I-component",
|
| 26 |
+
"3": "B-api_endpoint",
|
| 27 |
+
"4": "I-api_endpoint",
|
| 28 |
+
"5": "B-algorithm",
|
| 29 |
+
"6": "I-algorithm",
|
| 30 |
+
"7": "B-data_structure",
|
| 31 |
+
"8": "I-data_structure",
|
| 32 |
+
"9": "B-config",
|
| 33 |
+
"10": "I-config",
|
| 34 |
+
"11": "B-tool",
|
| 35 |
+
"12": "I-tool",
|
| 36 |
+
"13": "B-file_path",
|
| 37 |
+
"14": "I-file_path",
|
| 38 |
+
"15": "B-metric",
|
| 39 |
+
"16": "I-metric"
|
| 40 |
+
},
|
| 41 |
+
"label2id": {
|
| 42 |
+
"O": 0,
|
| 43 |
+
"B-component": 1,
|
| 44 |
+
"I-component": 2,
|
| 45 |
+
"B-api_endpoint": 3,
|
| 46 |
+
"I-api_endpoint": 4,
|
| 47 |
+
"B-algorithm": 5,
|
| 48 |
+
"I-algorithm": 6,
|
| 49 |
+
"B-data_structure": 7,
|
| 50 |
+
"I-data_structure": 8,
|
| 51 |
+
"B-config": 9,
|
| 52 |
+
"I-config": 10,
|
| 53 |
+
"B-tool": 11,
|
| 54 |
+
"I-tool": 12,
|
| 55 |
+
"B-file_path": 13,
|
| 56 |
+
"I-file_path": 14,
|
| 57 |
+
"B-metric": 15,
|
| 58 |
+
"I-metric": 16
|
| 59 |
+
},
|
| 60 |
+
"num_labels": 17,
|
| 61 |
+
"ultracode": {
|
| 62 |
+
"format": "raw-safetensors/1",
|
| 63 |
+
"exported_from": "model.onnx",
|
| 64 |
+
"onnx_sha256": "5a317a02eaccef10ed9bf494b768edfb64e651e112af90ec0a38e974b8d05269",
|
| 65 |
+
"head": "token_cls",
|
| 66 |
+
"position_ids": "roberta",
|
| 67 |
+
"pooling": "none: per-token classifier"
|
| 68 |
+
}
|
| 69 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8e96f32e7f39e1fa4d886807e668b02c8adc193d492b51d4e22b042af2eee22f
|
| 3 |
+
size 470021692
|