From 7dc5d8d06522b2ffb44876f2c6c8a974f27ff8a9 Mon Sep 17 00:00:00 2001 From: Lysandre Date: Tue, 30 Mar 2021 12:53:56 -0400 Subject: [PATCH] Sample --- config.json | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/config.json b/config.json index 593281a..f70148d 100644 --- a/config.json +++ b/config.json @@ -70,4 +70,10 @@ "vocab_size": 50257, "window_size": 256, "tokenizer_class": "GPT2Tokenizer" + "task_specific_params": { + "text-generation": { + "do_sample": true, + "max_length": 50 + } + } }