Does it support beam search generation strategy?

[colinzhaoxp]

I have a question about beam search as the title described. In other words, Does DiffuLLaMA support Beam search, Top-K, Top-p generation strategy that supported by AR models ?

[summmeer]

Hi, we implemented top-p during sampling:

DiffuLLaMA/LLaMA-Factory/src/llamafactory/train/ddm/trainer.py

Line 502 in 60abd1e

filter_logits = top_p_logits(logits/logits_temp, p=0.9)

Currently beam search of DLM is still under-exploration.