Diversity in beam search

#56
by ayseozgun - opened

Hello,

I want to create 5 different outputs by using flan-t5-xxl.
Here is my generate code:

"outputs = model.generate(input_ids, max_new_tokens=300,
num_beams=5,
num_return_sequences=5,
num_beam_groups=5,
diversity_penalty=5.0,
no_repeat_ngram_size=2)"
I read num_beam_groups and diversity_penalty parameters should affect the output diversity, but when I run this code, the outputs are still so similar each other.

Can you please help me?
Thanks in advance,

Sign up or log in to comment