Nemotron-Parse-2.0-8bit / auxiliary_prediction_heads.json
axiom-of-choice's picture
Add MLX conversion from nemotron-parse-2.0-8bit
e7e8947 verified
Raw
History Blame Contribute Delete
718 Bytes
{
"format": "safetensors",
"description": "Auxiliary decoder prediction-head tensors retained from training. These are not loaded by the default HF/vLLM inference path.",
"tensor_file": "auxiliary_prediction_heads.safetensors.extra",
"tensors": {
"decoder.extra_heads.0.bias": {
"shape": [
1024
],
"dtype": "float32"
},
"decoder.extra_heads.0.weight": {
"shape": [
1024,
1024
],
"dtype": "float32"
},
"decoder.extra_proj.0.bias": {
"shape": [
1024
],
"dtype": "float32"
},
"decoder.extra_proj.0.weight": {
"shape": [
1024,
1024
],
"dtype": "float32"
}
}
}