From 847d3703011fcf55b858ab7edcbff1a983715c05 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Mon, 9 Mar 2020 16:18:29 +0100 Subject: [PATCH] fix typo --- src/transformers/configuration_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/configuration_utils.py b/src/transformers/configuration_utils.py index 6db540ac0b..1ada6ade8b 100644 --- a/src/transformers/configuration_utils.py +++ b/src/transformers/configuration_utils.py @@ -70,7 +70,7 @@ class PretrainedConfig(object): # Parameters for sequence generation self.max_length = kwargs.pop("max_length", 20) self.do_sample = kwargs.pop("do_sample", False) - self.do_sample = kwargs.pop("early_stopping", False) + self.early_stopping = kwargs.pop("early_stopping", False) self.num_beams = kwargs.pop("num_beams", 1) self.temperature = kwargs.pop("temperature", 1.0) self.top_k = kwargs.pop("top_k", 50)