Views
No views yet
topic classification, sentiment analysis and as a reranker in RAG pipelines.1pip install gliclass
2pip install -U transformers>=4.48.01from gliclass import GLiClassModel, ZeroShotClassificationPipeline
2from transformers import AutoTokenizer
3
4model = GLiClassModel.from_pretrained("knowledgator/gliclass-modern-large-v2.0")
5tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-modern-large-v2.0", add_prefix_space=True)
6pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')
7
8text = "One day I will see the world!"
9labels = ["travel", "dreams", "sport", "science", "politics"]
10results = pipeline(text, labels, threshold=0.5)[0] #because we have one text
11for result in results:
12 print(result["label"], "=>", result["score"])1# Initialize model and multi-label pipeline
2text = "The cat slept on the windowsill all afternoon"
3labels = ["The cat was awake and playing outside."]
4results = pipeline(text, labels, threshold=0.0)[0]
5print(results)| Model | IMDB | AG_NEWS | Emotions |
|---|---|---|---|
| gliclass-modern-large-v2.0-init (399 M) | 0.9137 | 0.7357 | 0.4140 |
| gliclass-modern-base-v2.0-init (151 M) | 0.8264 | 0.6637 | 0.2985 |
| gliclass-modern-large-v2.0 (399 M) | 0.9448 | 0.736 | 0.4970 |
| gliclass-modern-base-v2.0 (151 M) | 0.9188 | 0.7089 | 0.4250 |
| gliclass-large-v1.0 (438 M) | 0.9404 | 0.7516 | 0.4874 |
| gliclass-base-v1.0 (186 M) | 0.8650 | 0.6837 | 0.4749 |
| gliclass-small-v1.0 (144 M) | 0.8650 | 0.6805 | 0.4664 |
| Bart-large-mnli (407 M) | 0.89 | 0.6887 | 0.3765 |
| Deberta-base-v3 (184 M) | 0.85 | 0.6455 | 0.5095 |
| Comprehendo (184M) | 0.90 | 0.7982 | 0.5660 |
| SetFit BAAI/bge-small-en-v1.5 (33.4M) | 0.86 | 0.5636 | 0.5754 |
| Dataset | gliclass-modern-base-v2.0 | gliclass-modern-large-v2.0 | gliclass-modern-base-v2.0-init | gliclass-modern-large-v2.0-init |
|---|---|---|---|---|
| CR | 0.8976 | 0.9198 | 0.9041 | 0.8980 |
| sst2 | 0.8525 | 0.9318 | 0.9011 | 0.9434 |
| sst5 | 0.2348 | 0.2147 | 0.1972 | 0.1123 |
| 20_news_groups | 0.351 | 0.3755 | 0.2448 | 0.2792 |
| spam | 0.483 | 0.6608 | 0.5074 | 0.6364 |
| financial_phrasebank | 0.3475 | 0.3157 | 0.2537 | 0.2562 |
| imdb | 0.9188 | 0.9448 | 0.8255 | 0.9137 |
| ag_news | 0.6835 | 0.7025 | 0.6050 | 0.6933 |
| emotion | 0.3925 | 0.4325 | 0.2474 | 0.3746 |
| cap_sotu | 0.3725 | 0.4157 | 0.2929 | 0.2919 |
| rotten_tomatoes | 0.6955 | 0.7357 | 0.6630 | 0.5928 |
| AVERAGE: | 0.5563 | 0.6045 | 0.5129 | 0.5447 |
1@misc{stepanov2025gliclassgeneralistlightweightmodel,
2 title={GLiClass: Generalist Lightweight Model for Sequence Classification Tasks},
3 author={Ihor Stepanov and Mykhailo Shtopko and Dmytro Vodianytskyi and Oleksandr Lukashov and Alexander Yavorskyi and Mykyta Yaroshenko},
4 year={2025},
5 eprint={2508.07662},
6 archivePrefix={arXiv},
7 primaryClass={cs.LG},
8 url={https://arxiv.org/abs/2508.07662},
9}