knowledgator
/

gliclass-llama-1.3B-v1.0

Zero-Shot Classification

text classification

small language models

sentiment analysis

Model card Files Files and versions Community

Ihor commited on Sep 10

Commit

1cc673b

•

1 Parent(s): 99a4a9a

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -40,8 +40,8 @@ Than you need to initialize a model and a pipeline:
 from gliclass import GLiClassModel, ZeroShotClassificationPipeline
 from transformers import AutoTokenizer
-model = GLiClassModel.from_pretrained("knowledgator/gliclass-large-v1.0-lw")
-tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-large-v1.0-lw")
 pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')

 from gliclass import GLiClassModel, ZeroShotClassificationPipeline
 from transformers import AutoTokenizer
+model = GLiClassModel.from_pretrained("knowledgator/gliclass-llama-1.3B-v1.0")
+tokenizer = AutoTokenizer.from_pretrained("knowledgator/gliclass-llama-1.3B-v1.0")
 pipeline = ZeroShotClassificationPipeline(model, tokenizer, classification_type='multi-label', device='cuda:0')