diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index f64deef..4cf183b 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -10,7 +10,7 @@ import numpy as np class GPT2Generator: - def __init__(self, generate_num=60, temperature=0.4, top_k=40, top_p=0.9): + def __init__(self, generate_num=75, temperature=0.4, top_k=40, top_p=0.9): self.generate_num=generate_num self.temp = temperature self.top_k = top_k