boris commited on
Commit
51e299d
1 Parent(s): 40cef17

New model from https://wandb.ai/wandb/huggingtweets/runs/23hnup9i

Browse files
README.md CHANGED
@@ -1,28 +1,27 @@
1
  ---
2
  language: en
3
- thumbnail: https://www.huggingtweets.com/nytimes/1606233617402/predictions.png
4
  tags:
5
  - huggingtweets
6
  widget:
7
  - text: "My dream is"
8
  ---
9
 
10
- <link rel="stylesheet" href="https://unpkg.com/@tailwindcss/typography@0.2.x/dist/typography.min.css">
11
-
12
- <style>
13
- @media (prefers-color-scheme: dark) {
14
- .prose { color: #E2E8F0 !important; }
15
- .prose h2, .prose h3, .prose a, .prose thead { color: #F7FAFC !important; }
16
- }
17
- </style>
18
-
19
- <section class='prose'>
20
-
21
- <div>
22
- <div style="width: 132px; height:132px; border-radius: 50%; background-size: cover; background-image: url('https://pbs.twimg.com/profile_images/1098244578472280064/gjkVMelR_400x400.png')">
23
- </div>
24
- <div style="margin-top: 8px; font-size: 19px; font-weight: 800">The New York Times 🤖 AI Bot </div>
25
- <div style="font-size: 15px; color: #657786">@nytimes bot</div>
26
  </div>
27
 
28
  I was made with [huggingtweets](https://github.com/borisdayma/huggingtweets).
@@ -35,62 +34,41 @@ The model uses the following pipeline.
35
 
36
  ![pipeline](https://github.com/borisdayma/huggingtweets/blob/master/img/pipeline.png?raw=true)
37
 
38
- To understand how the model was developed, check the [W&B report](https://app.wandb.ai/wandb/huggingtweets/reports/HuggingTweets-Train-a-model-to-generate-tweets--VmlldzoxMTY5MjI).
39
 
40
  ## Training data
41
 
42
- The model was trained on [@nytimes's tweets](https://twitter.com/nytimes).
43
-
44
- <table style='border-width:0'>
45
- <thead style='border-width:0'>
46
- <tr style='border-width:0 0 1px 0; border-color: #CBD5E0'>
47
- <th style='border-width:0'>Data</th>
48
- <th style='border-width:0'>Quantity</th>
49
- </tr>
50
- </thead>
51
- <tbody style='border-width:0'>
52
- <tr style='border-width:0 0 1px 0; border-color: #E2E8F0'>
53
- <td style='border-width:0'>Tweets downloaded</td>
54
- <td style='border-width:0'>3242</td>
55
- </tr>
56
- <tr style='border-width:0 0 1px 0; border-color: #E2E8F0'>
57
- <td style='border-width:0'>Retweets</td>
58
- <td style='border-width:0'>496</td>
59
- </tr>
60
- <tr style='border-width:0 0 1px 0; border-color: #E2E8F0'>
61
- <td style='border-width:0'>Short tweets</td>
62
- <td style='border-width:0'>4</td>
63
- </tr>
64
- <tr style='border-width:0'>
65
- <td style='border-width:0'>Tweets kept</td>
66
- <td style='border-width:0'>2742</td>
67
- </tr>
68
- </tbody>
69
- </table>
70
-
71
- [Explore the data](https://wandb.ai/wandb/huggingtweets/runs/2fvmvqnf/artifacts), which is tracked with [W&B artifacts](https://docs.wandb.com/artifacts) at every step of the pipeline.
72
 
73
  ## Training procedure
74
 
75
  The model is based on a pre-trained [GPT-2](https://huggingface.co/gpt2) which is fine-tuned on @nytimes's tweets.
76
 
77
- Hyperparameters and metrics are recorded in the [W&B training run](https://wandb.ai/wandb/huggingtweets/runs/118jzh5v) for full transparency and reproducibility.
78
-
79
- At the end of training, [the final model](https://wandb.ai/wandb/huggingtweets/runs/118jzh5v/artifacts) is logged and versioned.
80
 
81
- ## Intended uses & limitations
82
 
83
- ### How to use
84
 
85
  You can use this model directly with a pipeline for text generation:
86
 
87
- <pre><code><span style="color:#03A9F4">from</span> transformers <span style="color:#03A9F4">import</span> pipeline
88
- generator = pipeline(<span style="color:#FF9800">'text-generation'</span>,
89
- model=<span style="color:#FF9800">'huggingtweets/nytimes'</span>)
90
- generator(<span style="color:#FF9800">"My dream is"</span>, num_return_sequences=<span style="color:#8BC34A">5</span>)</code></pre>
91
-
 
92
 
93
- ### Limitations and bias
94
 
95
  The model suffers from [the same limitations and bias as GPT-2](https://huggingface.co/gpt2#limitations-and-bias).
96
 
@@ -100,14 +78,8 @@ In addition, the data present in the user's tweets further affects the text gene
100
 
101
  *Built by Boris Dayma*
102
 
103
- </section>
104
-
105
  [![Follow](https://img.shields.io/twitter/follow/borisdayma?style=social)](https://twitter.com/intent/follow?screen_name=borisdayma)
106
 
107
- <section class='prose'>
108
  For more details, visit the project repository.
109
- </section>
110
 
111
  [![GitHub stars](https://img.shields.io/github/stars/borisdayma/huggingtweets?style=social)](https://github.com/borisdayma/huggingtweets)
112
-
113
- <!--- random size file -->
1
  ---
2
  language: en
3
+ thumbnail: https://github.com/borisdayma/huggingtweets/blob/master/img/logo.png?raw=true
4
  tags:
5
  - huggingtweets
6
  widget:
7
  - text: "My dream is"
8
  ---
9
 
10
+ <div class="inline-flex flex-col" style="line-height: 1.5;">
11
+ <div class="flex">
12
+ <div
13
+ style="display:inherit; margin-left: 4px; margin-right: 4px; width: 92px; height:92px; border-radius: 50%; background-size: cover; background-image: url(&#39;https://pbs.twimg.com/profile_images/1098244578472280064/gjkVMelR_400x400.png&#39;)">
14
+ </div>
15
+ <div
16
+ style="display:none; margin-left: 4px; margin-right: 4px; width: 92px; height:92px; border-radius: 50%; background-size: cover; background-image: url(&#39;&#39;)">
17
+ </div>
18
+ <div
19
+ style="display:none; margin-left: 4px; margin-right: 4px; width: 92px; height:92px; border-radius: 50%; background-size: cover; background-image: url(&#39;&#39;)">
20
+ </div>
21
+ </div>
22
+ <div style="text-align: center; margin-top: 3px; font-size: 16px; font-weight: 800">🤖 AI BOT 🤖</div>
23
+ <div style="text-align: center; font-size: 16px; font-weight: 800">The New York Times</div>
24
+ <div style="text-align: center; font-size: 14px;">@nytimes</div>
 
25
  </div>
26
 
27
  I was made with [huggingtweets](https://github.com/borisdayma/huggingtweets).
34
 
35
  ![pipeline](https://github.com/borisdayma/huggingtweets/blob/master/img/pipeline.png?raw=true)
36
 
37
+ To understand how the model was developed, check the [W&B report](https://wandb.ai/wandb/huggingtweets/reports/HuggingTweets-Train-a-Model-to-Generate-Tweets--VmlldzoxMTY5MjI).
38
 
39
  ## Training data
40
 
41
+ The model was trained on tweets from The New York Times.
42
+
43
+ | Data | The New York Times |
44
+ | --- | --- |
45
+ | Tweets downloaded | 3250 |
46
+ | Retweets | 503 |
47
+ | Short tweets | 0 |
48
+ | Tweets kept | 2747 |
49
+
50
+ [Explore the data](https://wandb.ai/wandb/huggingtweets/runs/19jlccdf/artifacts), which is tracked with [W&B artifacts](https://docs.wandb.com/artifacts) at every step of the pipeline.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
 
52
  ## Training procedure
53
 
54
  The model is based on a pre-trained [GPT-2](https://huggingface.co/gpt2) which is fine-tuned on @nytimes's tweets.
55
 
56
+ Hyperparameters and metrics are recorded in the [W&B training run](https://wandb.ai/wandb/huggingtweets/runs/23hnup9i) for full transparency and reproducibility.
 
 
57
 
58
+ At the end of training, [the final model](https://wandb.ai/wandb/huggingtweets/runs/23hnup9i/artifacts) is logged and versioned.
59
 
60
+ ## How to use
61
 
62
  You can use this model directly with a pipeline for text generation:
63
 
64
+ ```python
65
+ from transformers import pipeline
66
+ generator = pipeline('text-generation',
67
+ model='huggingtweets/nytimes')
68
+ generator("My dream is", num_return_sequences=5)
69
+ ```
70
 
71
+ ## Limitations and bias
72
 
73
  The model suffers from [the same limitations and bias as GPT-2](https://huggingface.co/gpt2#limitations-and-bias).
74
 
78
 
79
  *Built by Boris Dayma*
80
 
 
 
81
  [![Follow](https://img.shields.io/twitter/follow/borisdayma?style=social)](https://twitter.com/intent/follow?screen_name=borisdayma)
82
 
 
83
  For more details, visit the project repository.
 
84
 
85
  [![GitHub stars](https://img.shields.io/github/stars/borisdayma/huggingtweets?style=social)](https://github.com/borisdayma/huggingtweets)
 
 
config.json CHANGED
@@ -8,7 +8,6 @@
8
  "bos_token_id": 50256,
9
  "embd_pdrop": 0.1,
10
  "eos_token_id": 50256,
11
- "gradient_checkpointing": false,
12
  "initializer_range": 0.02,
13
  "layer_norm_epsilon": 1e-05,
14
  "model_type": "gpt2",
@@ -19,6 +18,7 @@
19
  "n_layer": 12,
20
  "n_positions": 1024,
21
  "resid_pdrop": 0.1,
 
22
  "summary_activation": null,
23
  "summary_first_dropout": 0.1,
24
  "summary_proj_to_labels": true,
@@ -34,5 +34,8 @@
34
  "top_p": 0.95
35
  }
36
  },
 
 
 
37
  "vocab_size": 50257
38
  }
8
  "bos_token_id": 50256,
9
  "embd_pdrop": 0.1,
10
  "eos_token_id": 50256,
 
11
  "initializer_range": 0.02,
12
  "layer_norm_epsilon": 1e-05,
13
  "model_type": "gpt2",
18
  "n_layer": 12,
19
  "n_positions": 1024,
20
  "resid_pdrop": 0.1,
21
+ "scale_attn_weights": true,
22
  "summary_activation": null,
23
  "summary_first_dropout": 0.1,
24
  "summary_proj_to_labels": true,
34
  "top_p": 0.95
35
  }
36
  },
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.11.3",
39
+ "use_cache": true,
40
  "vocab_size": 50257
41
  }
flax_model.msgpack DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:5b72e020dd16b2ee5b5e3a9a34cb3299e36aaef5c694251c3c5b5f32ce7e1f04
3
- size 497764120
 
 
 
merges.txt CHANGED
@@ -1,4 +1,4 @@
1
- #version: 0.2
2
  Ġ t
3
  Ġ a
4
  h e
1
+ #version: 0.2 - Trained by `huggingface/tokenizers`
2
  Ġ t
3
  Ġ a
4
  h e
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f71d666fdf9c5567b521e7fb9f2e35d94576f488d49c735a0654492010a11071
3
- size 510406549
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff499736245e605199e3f08bd54334400f80d6e01abce9382502a7f3108ead3e
3
+ size 510403817
special_tokens_map.json CHANGED
@@ -1 +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}}
1
+ {"bos_token": "<|endoftext|>", "eos_token": "<|endoftext|>", "unk_token": "<|endoftext|>"}
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
tokenizer_config.json CHANGED
@@ -1 +1 @@
1
- {"errors": "replace", "unk_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "bos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "eos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "add_prefix_space": false, "model_max_length": 1024, "name_or_path": "gpt2"}
1
+ {"unk_token": "<|endoftext|>", "bos_token": "<|endoftext|>", "eos_token": "<|endoftext|>", "add_prefix_space": false, "model_max_length": 1024, "special_tokens_map_file": null, "name_or_path": "gpt2", "tokenizer_class": "GPT2Tokenizer"}
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:dd18ee94d900e1b13ae77bc602ab644a5b8a5095d052c81c2f4f8bd339530b50
3
- size 1775
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08460aa6cbc9bc56b765719a206173329dc21152daeab21a0a8ece24a5a6ea96
3
+ size 2863
vocab.json CHANGED
The diff for this file is too large to render. See raw diff