New model from https://wandb.ai/wandb/huggingtweets/runs/3nxbm83r

README.md

@@ -1,28 +1,17 @@
 ---
 language: en
-thumbnail: https://www.huggingtweets.com/stephenking/1604092123749/predictions.png
+thumbnail: https://github.com/borisdayma/huggingtweets/blob/master/img/logo.png?raw=true
 tags:
 - huggingtweets
 widget:
 - text: "My dream is"
 ---
 
-<link rel="stylesheet" href="https://unpkg.com/@tailwindcss/typography@0.2.x/dist/typography.min.css">
-
-<style>
-@media (prefers-color-scheme: dark) {
-  .prose { color: #E2E8F0 !important; }
-  .prose h2, .prose h3, .prose a, .prose thead { color: #F7FAFC !important; }
-}
-</style>
-
-<section class='prose'>
-
 <div>
 <div style="width: 132px; height:132px; border-radius: 50%; background-size: cover; background-image: url('https://pbs.twimg.com/profile_images/378800000836981162/b683f7509ec792c3e481ead332940cdc_400x400.jpeg')">
 </div>
 <div style="margin-top: 8px; font-size: 19px; font-weight: 800">Stephen King 🤖 AI Bot </div>
-<div style="font-size: 15px; color: #657786">@stephenking bot</div>
+<div style="font-size: 15px">@stephenking bot</div>
 </div>
 
 I was made with [huggingtweets](https://github.com/borisdayma/huggingtweets).
@@ -41,56 +30,35 @@ To understand how the model was developed, check the [W&B report](https://app.wa
 
 The model was trained on [@stephenking's tweets](https://twitter.com/stephenking).
 
-<table style='border-width:0'>
-<thead style='border-width:0'>
-<tr style='border-width:0 0 1px 0; border-color: #CBD5E0'>
-<th style='border-width:0'>Data</th>
-<th style='border-width:0'>Quantity</th>
-</tr>
-</thead>
-<tbody style='border-width:0'>
-<tr style='border-width:0 0 1px 0; border-color: #E2E8F0'>
-<td style='border-width:0'>Tweets downloaded</td>
-<td style='border-width:0'>3194</td>
-</tr>
-<tr style='border-width:0 0 1px 0; border-color: #E2E8F0'>
-<td style='border-width:0'>Retweets</td>
-<td style='border-width:0'>747</td>
-</tr>
-<tr style='border-width:0 0 1px 0; border-color: #E2E8F0'>
-<td style='border-width:0'>Short tweets</td>
-<td style='border-width:0'>178</td>
-</tr>
-<tr style='border-width:0'>
-<td style='border-width:0'>Tweets kept</td>
-<td style='border-width:0'>2269</td>
-</tr>
-</tbody>
-</table>
-
-[Explore the data](https://app.wandb.ai/wandb/huggingtweets/runs/ey1pt4eu/artifacts), which is tracked with [W&B artifacts](https://docs.wandb.com/artifacts) at every step of the pipeline.
+| Data | Quantity |
+| --- | --- |
+| Tweets downloaded | 3182 |
+| Retweets | 721 |
+| Short tweets | 189 |
+| Tweets kept | 2272 |
+
+[Explore the data](https://wandb.ai/wandb/huggingtweets/runs/26bv6b2f/artifacts), which is tracked with [W&B artifacts](https://docs.wandb.com/artifacts) at every step of the pipeline.
 
 ## Training procedure
 
 The model is based on a pre-trained [GPT-2](https://huggingface.co/gpt2) which is fine-tuned on @stephenking's tweets.
 
-Hyperparameters and metrics are recorded in the [W&B training run](https://app.wandb.ai/wandb/huggingtweets/runs/cgu92b3q) for full transparency and reproducibility.
+Hyperparameters and metrics are recorded in the [W&B training run](https://wandb.ai/wandb/huggingtweets/runs/3nxbm83r) for full transparency and reproducibility.
 
-At the end of training, [the final model](https://app.wandb.ai/wandb/huggingtweets/runs/cgu92b3q/artifacts) is logged and versioned.
+At the end of training, [the final model](https://wandb.ai/wandb/huggingtweets/runs/3nxbm83r/artifacts) is logged and versioned.
 
-## Intended uses & limitations
-
-### How to use
+## How to use
 
 You can use this model directly with a pipeline for text generation:
 
-<pre><code><span style="color:#03A9F4">from</span> transformers <span style="color:#03A9F4">import</span> pipeline
-generator = pipeline(<span style="color:#FF9800">'text-generation'</span>,
-                     model=<span style="color:#FF9800">'huggingtweets/stephenking'</span>)
-generator(<span style="color:#FF9800">"My dream is"</span>, num_return_sequences=<span style="color:#8BC34A">5</span>)</code></pre>
-
+```python
+from transformers import pipeline
+generator = pipeline('text-generation',
+                     model='huggingtweets/stephenking')
+generator("My dream is", num_return_sequences=5)
+```
 
-### Limitations and bias
+## Limitations and bias
 
 The model suffers from [the same limitations and bias as GPT-2](https://huggingface.co/gpt2#limitations-and-bias).
 
@@ -100,14 +68,8 @@ In addition, the data present in the user's tweets further affects the text gene
 
 *Built by Boris Dayma*
 
-</section>
-
 [![Follow](https://img.shields.io/twitter/follow/borisdayma?style=social)](https://twitter.com/intent/follow?screen_name=borisdayma)
 
-<section class='prose'>
 For more details, visit the project repository.
-</section>
 
 [![GitHub stars](https://img.shields.io/github/stars/borisdayma/huggingtweets?style=social)](https://github.com/borisdayma/huggingtweets)
-
-<!--- random size file                                                                                 -->

config.json

@@ -1,4 +1,5 @@
 {
+  "_name_or_path": "gpt2",
   "activation_function": "gelu_new",
   "architectures": [
     "GPT2LMHeadModel"
@@ -7,6 +8,7 @@
   "bos_token_id": 50256,
   "embd_pdrop": 0.1,
   "eos_token_id": 50256,
+  "gradient_checkpointing": false,
   "initializer_range": 0.02,
   "layer_norm_epsilon": 1e-05,
   "model_type": "gpt2",
@@ -32,5 +34,7 @@
       "top_p": 0.95
     }
   },
+  "transformers_version": "4.3.2",
+  "use_cache": true,
   "vocab_size": 50257
 }

merges.txt

@@ -1,4 +1,4 @@
-#version: 0.2
+#version: 0.2 - Trained by `huggingface/tokenizers`
 Ġ t
 Ġ a
 h e

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ee554f18a5487a0a31e7fae72de5f7e61c58b41407d4badf83e4edbfcf9fed3d
-size 510406559
+oid sha256:c9e0218fc5efaae08967fb1e6c46ee4f76a6af8f029484ada436ab91e93d2b26
+size 510406805

special_tokens_map.json

@@ -1 +1 @@
-{"bos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "eos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "unk_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}}
+{"bos_token": "<|endoftext|>", "eos_token": "<|endoftext|>", "unk_token": "<|endoftext|>"}

tokenizer_config.json

@@ -1 +1 @@
-{"model_max_length": 1024}
+{"unk_token": "<|endoftext|>", "bos_token": "<|endoftext|>", "eos_token": "<|endoftext|>", "add_prefix_space": false, "model_max_length": 1024, "name_or_path": "gpt2"}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5896b42542cb6da662ceb0cd729be2ffc5ad0aaa3d804007756df6e5d79ffe3a
+size 2159