Instructions to use dacorvo/tiny-random-gpt2-neuronx-no-checkpoint with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dacorvo/tiny-random-gpt2-neuronx-no-checkpoint with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("dacorvo/tiny-random-gpt2-neuronx-no-checkpoint", dtype="auto") - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files- config.json +1 -1
config.json
CHANGED
|
@@ -23,7 +23,7 @@
|
|
| 23 |
"auto_cast_type": "fp32",
|
| 24 |
"batch_size": 16,
|
| 25 |
"checkpoint_id": "hf-internal-testing/tiny-random-gpt2",
|
| 26 |
-
"checkpoint_revision":
|
| 27 |
"compiler_type": "neuronx-cc",
|
| 28 |
"compiler_version": "2.12.54.0+f631c2365",
|
| 29 |
"num_cores": 2,
|
|
|
|
| 23 |
"auto_cast_type": "fp32",
|
| 24 |
"batch_size": 16,
|
| 25 |
"checkpoint_id": "hf-internal-testing/tiny-random-gpt2",
|
| 26 |
+
"checkpoint_revision": "91c0fe31d692dd8448d9bc06e8d1877345009e3b",
|
| 27 |
"compiler_type": "neuronx-cc",
|
| 28 |
"compiler_version": "2.12.54.0+f631c2365",
|
| 29 |
"num_cores": 2,
|