Instructions to use optimum-internal-testing/tiny_random_bert_neuronx with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use optimum-internal-testing/tiny_random_bert_neuronx with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="optimum-internal-testing/tiny_random_bert_neuronx")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("optimum-internal-testing/tiny_random_bert_neuronx") model = AutoModel.from_pretrained("optimum-internal-testing/tiny_random_bert_neuronx", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload config.json with huggingface_hub
Browse files- config.json +1 -0
config.json
CHANGED
|
@@ -28,6 +28,7 @@
|
|
| 28 |
"attention_mask",
|
| 29 |
"token_type_ids"
|
| 30 |
],
|
|
|
|
| 31 |
"int_dtype": "int64",
|
| 32 |
"model_type": "bert",
|
| 33 |
"optlevel": "2",
|
|
|
|
| 28 |
"attention_mask",
|
| 29 |
"token_type_ids"
|
| 30 |
],
|
| 31 |
+
"instance_type": "inf2",
|
| 32 |
"int_dtype": "int64",
|
| 33 |
"model_type": "bert",
|
| 34 |
"optlevel": "2",
|