Downloads · 30 days
9
32% of all-time downloads
optimum/bert-base-multilingual-uncased-sentiment-neuronx
bert-base-multilingual-uncased-sentiment-neuronx is a machine learning model from optimum. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
Downloads · 30 days
9
32% of all-time downloads
All-time downloads
28
Public
Repo size
442 MB
Likes
0
Public
Click a slice to open those files.
.neuron442 MB · 99%
From the Hugging Face model README
INFERENTIA ONLY
from transformers import AutoTokenizer
from optimum.neuron import NeuronBertForSequenceClassification
input_shapes = {"batch_size": 1, "sequence_length": 128}
compiler_args = {"auto_cast": "matmul", "auto_cast_type": "bf16"}
neuron_model = NeuronBertForSequenceClassification.from_pretrained(
"nlptown/bert-base-multilingual-uncased-sentiment",
export=True,
**input_shapes,
**compiler_args,
)
# Save locally
neuron_model.save_pretrained("bert_base_multilingual_uncased_sentiment_neuronx")
neuron_model.push_to_hub(
"bert_base_multilingual_uncased_sentiment_neuronx",
repository_id="optimum/bert-base-multilingual-uncased-sentiment-neuronx", # Replace with your HF Hub repo id
)