Instructions to use typeform/squeezebert-mnli with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use typeform/squeezebert-mnli with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("zero-shot-classification", model="typeform/squeezebert-mnli")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("typeform/squeezebert-mnli", dtype="auto") - Notebooks
- Google Colab
- Kaggle
David Chu commited on
Commit ·
040a407
1
Parent(s): ba78a32
Update config.json
Browse files- config.json +4 -4
config.json
CHANGED
|
@@ -20,13 +20,13 @@
|
|
| 20 |
"output_groups": 4,
|
| 21 |
"num_labels": 3,
|
| 22 |
"label2id": {
|
| 23 |
-
"ENTAILMENT":
|
| 24 |
-
"NEUTRAL":
|
| 25 |
"CONTRADICTION": 0
|
| 26 |
},
|
| 27 |
"id2label": {
|
| 28 |
-
"
|
| 29 |
-
"
|
| 30 |
"0": "CONTRADICTION"
|
| 31 |
}
|
| 32 |
}
|
|
|
|
| 20 |
"output_groups": 4,
|
| 21 |
"num_labels": 3,
|
| 22 |
"label2id": {
|
| 23 |
+
"ENTAILMENT": 2,
|
| 24 |
+
"NEUTRAL": 1,
|
| 25 |
"CONTRADICTION": 0
|
| 26 |
},
|
| 27 |
"id2label": {
|
| 28 |
+
"2": "ENTAILMENT",
|
| 29 |
+
"1": "NEUTRAL",
|
| 30 |
"0": "CONTRADICTION"
|
| 31 |
}
|
| 32 |
}
|