Instructions to use baseten/gemma-4-e2b-it-sequence-classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use baseten/gemma-4-e2b-it-sequence-classification with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="baseten/gemma-4-e2b-it-sequence-classification", trust_remote_code=True)# Load model directly from transformers import AutoProcessor, AutoModelForSequenceClassification processor = AutoProcessor.from_pretrained("baseten/gemma-4-e2b-it-sequence-classification", trust_remote_code=True) model = AutoModelForSequenceClassification.from_pretrained("baseten/gemma-4-e2b-it-sequence-classification", trust_remote_code=True) - Notebooks
- Google Colab
- Kaggle
Update config.json
Browse files- config.json +4 -4
config.json
CHANGED
|
@@ -48,10 +48,10 @@
|
|
| 48 |
},
|
| 49 |
"boa_token_id": 256000,
|
| 50 |
"boi_token_id": 255999,
|
| 51 |
-
"
|
| 52 |
-
|
| 53 |
-
|
| 54 |
-
|
| 55 |
"dtype": "bfloat16",
|
| 56 |
"eoa_token_id": 258883,
|
| 57 |
"eoa_token_index": 258883,
|
|
|
|
| 48 |
},
|
| 49 |
"boa_token_id": 256000,
|
| 50 |
"boi_token_id": 255999,
|
| 51 |
+
"id2label": {"0": "no", "1": "yes"},
|
| 52 |
+
"label2id": {"no": 0, "yes": 1},
|
| 53 |
+
"classifier_token_ids": {"no": 1904, "yes": 4443},
|
| 54 |
+
"max_position_embeddings": 26214,
|
| 55 |
"dtype": "bfloat16",
|
| 56 |
"eoa_token_id": 258883,
|
| 57 |
"eoa_token_index": 258883,
|