Modelpublic

MadjidKrb/qwen3-slm-text-classifier

sourceHugging Faceapache-2.0updated 8mo agoView on Hugging Face

0likes6downloads

Model Card

Text Classification Model

Ce modèle est un classifier basé sur Qwen3-0.6B avec LoRA fine-tuning.

Utilisation

python

from transformers import AutoTokenizer, AutoModel
from peft import PeftModel
import torch

# Charger le tokenizer
tokenizer = AutoTokenizer.from_pretrained("MadjidKrb/qwen3-slm-text-classifier")

# Charger le modèle de base
base_model = AutoModel.from_pretrained("Qwen/Qwen3-0.6B", trust_remote_code=True)

# Charger les poids LoRA
model = PeftModel.from_pretrained(base_model, "MadjidKrb/qwen3-slm-text-classifier/lora_weights")

Détails

—Modèle de base: Qwen/Qwen3-0.6B
—Nombre de classes: 4
—Méthode: LoRA fine-tuning
—Classes: ['World', 'Sports', 'Business', 'Sci/Tech']