From c76450e20c052d63f8d50ab9d68bd2d7d96a3e48 Mon Sep 17 00:00:00 2001 From: Oleksandr Bushkovskyi Date: Wed, 20 May 2020 16:44:26 +0300 Subject: [PATCH] Model card for Tereveni-AI/gpt2-124M-uk-fiction (#4470) Create model card for "Tereveni-AI/gpt2-124M-uk-fiction" model --- .../gpt2-124M-uk-fiction/README.md | 23 +++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 model_cards/Tereveni-AI/gpt2-124M-uk-fiction/README.md diff --git a/model_cards/Tereveni-AI/gpt2-124M-uk-fiction/README.md b/model_cards/Tereveni-AI/gpt2-124M-uk-fiction/README.md new file mode 100644 index 00000000..1e24387c --- /dev/null +++ b/model_cards/Tereveni-AI/gpt2-124M-uk-fiction/README.md @@ -0,0 +1,23 @@ +Note: **default code snippet above won't work** because we are using `AlbertTokenizer` with `GPT2LMHeadModel`, see [issue](https://github.com/huggingface/transformers/issues/4285). + +## GPT2 124M Trained on Ukranian Fiction + +Example usage: +```python +from transformers import AlbertTokenizer, GPT2LMHeadModel + +tokenizer = AlbertTokenizer.from_pretrained("Tereveni-AI/gpt2-124M-uk-fiction") +model = GPT2LMHeadModel.from_pretrained("Tereveni-AI/gpt2-124M-uk-fiction") + +input_ids = tokenizer.encode('Но зла Юнона, суча дочка,', add_special_tokens=False, return_tensors='pt') + +outputs = model.generate( + input_ids, + do_sample=True, + num_return_sequences=3, + max_length=50 +) + +for i, out in enumerate(outputs): + print('{}: {}'.format(i, tokenizer.decode(out))) +```