Text Generation
PyTorch
Safetensors
English
gpt2
nielsr HF Staff commited on
Commit
ee9a49a
·
verified ·
1 Parent(s): cef47b6

Add `library_name: transformers` metadata tag

Browse files

This PR improves the model card by adding the `library_name: transformers` metadata tag. This addition is based on the `config.json` and `tokenizer_config.json` files, which indicate the model's compatibility with the Hugging Face Transformers library (e.g., `GPT2LMHeadModel`, `GPT2Tokenizer`, `transformers_version`).

Adding this tag will enable the automated "How to use" widget on the model page, providing users with a convenient and predefined code snippet for loading and utilizing the model with the 🤗 Transformers library.

Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -1,15 +1,17 @@
1
  ---
2
- license: apache-2.0
 
3
  datasets:
4
  - databricks/databricks-dolly-15k
5
  language:
6
  - en
 
7
  metrics:
8
  - rouge
9
- base_model:
10
- - openai-community/gpt2
11
  pipeline_tag: text-generation
 
12
  ---
 
13
  # init-gpt2-120M
14
 
15
  [paper](https://arxiv.org/abs/2306.08543) | [code](https://github.com/microsoft/LMOps/tree/main/minillm)
 
1
  ---
2
+ base_model:
3
+ - openai-community/gpt2
4
  datasets:
5
  - databricks/databricks-dolly-15k
6
  language:
7
  - en
8
+ license: apache-2.0
9
  metrics:
10
  - rouge
 
 
11
  pipeline_tag: text-generation
12
+ library_name: transformers
13
  ---
14
+
15
  # init-gpt2-120M
16
 
17
  [paper](https://arxiv.org/abs/2306.08543) | [code](https://github.com/microsoft/LMOps/tree/main/minillm)