Downloads · 30 days
14
5% of all-time downloads
bert-base-NER-neuronx
optimum/bert-base-NER-neuronx
bert-base-NER-neuronx is a token classification model from optimum. Use it when you need labels on individual words, such as names. It is set up for transformers. The card lists the license as mit.
All-time downloads
273
Public
Repo size
621 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.neuron220 MB · 100%
At a glance
- Task
- Token Classification
- Library
- transformers
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- Jun 6, 2023
- Updated
- Apr 2, 2025
- SHA
- 78983a14
Try a prompt
Model card
From the Hugging Face model README
INFERENTIA ONLY
from transformers import AutoTokenizer
from optimum.neuron import NeuronBertForTokenClassification
input_shapes = {"batch_size": 1, "sequence_length": 128}
compiler_args = {"auto_cast": "matmul", "auto_cast_type": "bf16"}
neuron_model = NeuronBertForTokenClassification.from_pretrained(
"dslim/bert-base-NER",
export=True,
**input_shapes,
**compiler_args,
)
# Save locally
neuron_model.save_pretrained("bert_base_ner_neuronx")
neuron_model.push_to_hub(
"bert_base_ner_neuronx",
repository_id="optimum/bert-base-NER-neuronx", # Replace with your HF Hub repo id
)
- Task
- Token Classification
- Library
- transformers
- Type
- bert
- License
- mit
- Created
- Jun 6, 2023
- Updated
- Apr 2, 2025