from optimum.neuron import NeuronModelForSequenceClassification

model = NeuronModelForSequenceClassification.from_pretrained("badaoui/oxyapi-albert-moderation-001-neuron")

Performance

Base Model: <a href="https://huggingface.co/oxyapi/albert-moderation-001" target="_blank" rel="noopener noreferrer" class="text-blue-400 hover:text-blue-300 underline underline-offset-2 transition-colors">oxyapi/albert-moderation-001
Task: text-classification
Optimization: AWS Neuron compilation
Generated by: <a href="https://huggingface.co/badaoui" target="_blank" rel="noopener noreferrer" class="text-blue-400 hover:text-blue-300 underline underline-offset-2 transition-colors">badaoui
Generated using: <a href="https://huggingface.co/spaces/optimum/neuron-export" target="_blank" rel="noopener noreferrer" class="text-blue-400 hover:text-blue-300 underline underline-offset-2 transition-colors">Optimum Neuron Compiler Space

These files are pre-compiled for AWS Neuron devices and should provide improved inference performance compared to the original model when deployed on Inferentia or Trainium instances.

Original Model

For the original model, training details, and more information, please visit: oxyapi/albert-moderation-001