Commit 847d3703 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

fix typo

parent 3e624c64
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment