From 18411abee94b53617fa9d92969ef40169961c89f Mon Sep 17 00:00:00 2001 From: system Date: Thu, 7 May 2020 02:25:31 +0000 Subject: [PATCH] Update config.json --- config.json | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/config.json b/config.json index c5fd6d2..5da9a52 100644 --- a/config.json +++ b/config.json @@ -8,6 +8,11 @@ "MarianMTModel" ], "attention_dropout": 0.0, + "bad_words_ids": [ + [ + 58378 + ] + ], "bos_token_id": 0, "classif_dropout": 0.0, "d_model": 512, @@ -34,6 +39,7 @@ "LABEL_1": 1, "LABEL_2": 2 }, + "max_length": 512, "max_position_embeddings": 512, "model_type": "marian", "normalize_before": false,