Text Generation
PyTorch
English
gpt2
nielsr HF Staff commited on
Commit
becc585
·
verified ·
1 Parent(s): 090b000

Add `library_name: transformers` to model card metadata

Browse files

This PR improves the model card by adding the `library_name: transformers` tag to the metadata. 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.

Evidence from `config.json` and `tokenizer_config.json` indicates full compatibility with `transformers`, showcasing `GPT2LMHeadModel` architecture and `GPT2Tokenizer`.

Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -1,14 +1,15 @@
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-medium
11
  pipeline_tag: text-generation
 
12
  ---
13
 
14
  # MiniLLM-gpt2-340M
 
1
  ---
2
+ base_model:
3
+ - openai-community/gpt2-medium
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-340M