Modelpublic

badaoui/oxyapi-albert-moderation-001-neuron

sourceHugging Faceupdated 1y agoView on Hugging Face

0likes3downloads

Model Card

Neuron-Optimized oxyapi/albert-moderation-001

This repository contains AWS Neuron-optimized files for oxyapi/albert-moderation-001.

Model Details

—Base Model: oxyapi/albert-moderation-001
—Task: text-classification
—Optimization: AWS Neuron compilation
—Generated by: badaoui
—Generated using: Optimum Neuron Compiler Space

Usage

This model has been optimized for AWS Neuron devices (Inferentia/Trainium). To use it:

python

from optimum.neuron import NeuronModelForSequenceClassification

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

Performance

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