Instructions to use pravinbhat1/distilgpt2 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use pravinbhat1/distilgpt2 with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("pravinbhat1/distilgpt2", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Create config.json
Browse files- config.json +6 -0
config.json
ADDED
|
@@ -0,0 +1,6 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"do_sample": true,
|
| 3 |
+
"eos_token_id": 50256,
|
| 4 |
+
"max_new_tokens": 50,
|
| 5 |
+
"transformers_version": "4.34.1"
|
| 6 |
+
}
|