metadata
tags:
- neuron
- optimized
- aws-neuron
- text-classification
base_model: DAMO-NLP-SG/zero-shot-classify-SSTuning-ALBERT
Neuron-Optimized DAMO-NLP-SG/zero-shot-classify-SSTuning-ALBERT
This repository contains AWS Neuron-optimized files for DAMO-NLP-SG/zero-shot-classify-SSTuning-ALBERT.
Model Details
- Base Model: DAMO-NLP-SG/zero-shot-classify-SSTuning-ALBERT
- 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:
from optimum.neuron import NeuronModelForSequenceClassification
model = NeuronModelForSequenceClassification.from_pretrained("badaoui/DAMO-NLP-SG-zero-shot-classify-SSTuning-ALBERT-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: DAMO-NLP-SG/zero-shot-classify-SSTuning-ALBERT