olm
/

olm-gpt2-oct-2022

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

Tristan commited on Dec 20, 2022

Commit

510e07a

·

1 Parent(s): f2ae817

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -26,7 +26,7 @@ set a seed for reproducibility:
 >>> # It is important to include the bad_words_ids=[[0,2]] if you want this model to stay on topic.
 >>> # Otherwise, the model may generate start and end tokens followed by text that is not relevant to
 >>> # the previous text.
->>> generator = pipeline('text-generation', model='olm/olm-gpt2-oct-2022')
 >>> set_seed(42)
 >>> # This example also illustrates that sometimes our model generates
 >>> # bloggy/spammy/webb-y things, even though it gets higher evaluation results

 >>> # It is important to include the bad_words_ids=[[0,2]] if you want this model to stay on topic.
 >>> # Otherwise, the model may generate start and end tokens followed by text that is not relevant to
 >>> # the previous text.
+>>> generator = pipeline('text-generation', model='olm/olm-gpt2-oct-2022', bad_words_ids=[[0,2]])
 >>> set_seed(42)
 >>> # This example also illustrates that sometimes our model generates
 >>> # bloggy/spammy/webb-y things, even though it gets higher evaluation results