generation_kwargs = {
    "do_sample": True,
    "top_k": 50,
    # "temperature": 0.8,
    "top_p": 0.95,
    # "early_stopping": True,
    # "num_beams": 10,
    "num_return_sequences": 3,
    # "no_repeat_ngram_size": 3,
    "max_new_tokens": 128,
}
