From 141727c63c9bac55dc66ae21d9036b715912674e Mon Sep 17 00:00:00 2001 From: Zhengbao Jiang Date: Wed, 26 Oct 2022 20:36:18 +0000 Subject: [PATCH] Add no_repeat_ngram_size to config to avoid repeatedly generating . --- config.json | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/config.json b/config.json index 774ae45..6460ba6 100644 --- a/config.json +++ b/config.json @@ -33,5 +33,6 @@ "torch_dtype": "float32", "transformers_version": "4.17.0.dev0", "use_cache": true, - "vocab_size": 50265 + "vocab_size": 50265, + "no_repeat_ngram_size": 3 }