PolyAI/banking77
Updated • 14.4k • 84
How to use hazemgalal1/gptneo125m-banking-complaint-classifier with PEFT:
from peft import PeftModel
from transformers import AutoModelForSequenceClassification
base_model = AutoModelForSequenceClassification.from_pretrained("EleutherAI/gpt-neo-125M")
model = PeftModel.from_pretrained(base_model, "hazemgalal1/gptneo125m-banking-complaint-classifier")| Setting | Value |
|---|---|
| Base Model | EleutherAI/gpt-neo-125M |
| LoRA Rank | 8 |
| Num Labels | 77 |
| Epochs | 5 |
| Learning Rate | 2e-4 |
For educational purposes only.