Text Generation
PyTorch
English
gpt2
nielsr HF Staff commited on
Commit
bcfb634
·
verified ·
1 Parent(s): 1eb1bba

Add `library_name: transformers` to metadata

Browse files

This PR improves the model card by adding the `library_name: transformers` metadata tag. This will enable the automated "How to use" widget on the Hugging Face Hub, providing users with a quick and easy way to interact with the model using the `transformers` library.

The model's compatibility with the `transformers` library is confirmed by the `config.json` (specifying `"model_type": "gpt2"` and `"architectures": ["GPT2LMHeadModel"]`) and `tokenizer_config.json` (specifying `"tokenizer_class": "GPT2Tokenizer"`) files.

Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -1,17 +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-large
11
  pipeline_tag: text-generation
 
12
  ---
13
 
14
-
15
  # MiniLLM-gpt2-760M
16
 
17
  [paper](https://arxiv.org/abs/2306.08543) | [code](https://github.com/microsoft/LMOps/tree/main/minillm)
 
1
  ---
2
+ base_model:
3
+ - openai-community/gpt2-large
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
  # MiniLLM-gpt2-760M
16
 
17
  [paper](https://arxiv.org/abs/2306.08543) | [code](https://github.com/microsoft/LMOps/tree/main/minillm)