Pythia Quantized Model for Sentiment Analysis
This repository hosts a quantized version of the Pythia model, fine-tuned for sentiment analysis tasks.
The model has been optimized for efficient deployment while maintaining high accuracy, making it suitable
for resource-constrained environments.
Model Details
-------------
* **Developed By:** AventIQ-AI
* **Model Architecture:** Pythia-410m
* **Task:** Sentiment Analysis
* **Dataset:** IMDb Reviews
* **Quantization:** Fp16
* **Fine-tuning Framework:** Hugging Face Transformers
The quantized model achieves comparable performance to the full-precision model while reducing memory usage and inference time.
Usage
-----
### Installation
pip install transformers torch
### Loading the Model
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("AventIQ-AI/pythia-410m")
model = AutoModelForSequenceClassification.from_pretrained("AventIQ-AI…
