trl-internal-testing
/

dummy-GPT2-correct-vocab

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

qgallouedec HF staff commited on Aug 8

Commit

e2df214

•

1 Parent(s): c5e4f2e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -8,7 +8,7 @@ tags: []
 ```python
 from transformers import AutoTokenizer, GPT2Config, GPT2LMHeadModel
-config = GPT2Config(n_positions=512, n_embd=32, n_layer=5, n_head=4, n_inner=37, is_decoder=True)
 model = GPT2LMHeadModel(config)
 tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")

 ```python
 from transformers import AutoTokenizer, GPT2Config, GPT2LMHeadModel
+config = GPT2Config(n_positions=512, n_embd=32, n_layer=5, n_head=4, n_inner=37, pad_token_id=1023, is_decoder=True)
 model = GPT2LMHeadModel(config)
 tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")