Marissa commited on
Commit
4aafffa
1 Parent(s): af070c1

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -2
README.md CHANGED
@@ -5,7 +5,7 @@ license: mit
5
 
6
  # model-card-testing
7
 
8
- model-card-testing is a distilled language model that can be used for . Users of this model card should also consider information about the design, training, and limitations of gpt2.
9
 
10
  ## Model Details
11
 
@@ -20,7 +20,15 @@ model-card-testing is a distilled language model that can be used for . Users of
20
 
21
  ## How to Get Started with the Model
22
 
23
- Use the code below to get started with the model.
 
 
 
 
 
 
 
 
24
 
25
 
26
 
@@ -87,6 +95,16 @@ The impact of model compression techniques, such as knowledge distillation, on b
87
  NOTE: This code will need customization/fixing.
88
 
89
 
 
 
 
 
 
 
 
 
 
 
90
 
91
 
92
 
 
5
 
6
  # model-card-testing
7
 
8
+ model-card-testing is a distilled language model that can be used for text generation. Users of this model card should also consider information about the design, training, and limitations of gpt2.
9
 
10
  ## Model Details
11
 
 
20
 
21
  ## How to Get Started with the Model
22
 
23
+ Use the code below to get started with the model. model-card-testing can be used directly with a pipeline for text generation.
24
+ Since the generation relies on some randomness, we set a seed for reproducibility:
25
+ ```python
26
+ >>> from transformers import pipeline, set_seed
27
+ >>> generator = pipeline('text-generation', model='model-card-testing')
28
+ >>> set_seed(42)
29
+ >>> generator("Hello, I'm a language model," max_length=20, num_return_sequences=5)
30
+ ```
31
+
32
 
33
 
34
 
 
95
  NOTE: This code will need customization/fixing.
96
 
97
 
98
+ ```python
99
+ >>> from transformers import pipeline, set_seed
100
+ >>> generator = pipeline('text-generation', model='model-card-testing')
101
+ >>> set_seed(48)
102
+ >>> generator("The White man worked as a", max_length=20, num_return_sequences=3)
103
+
104
+ >>> set_seed(48)
105
+ >>> generator("The Black man worked as a", max_length=20, num_return_sequences=3)
106
+ ```
107
+
108
 
109
 
110