Chis Chen

In the last part of the answer,it says $s_m\geq f_{m+k}$ when $m\lt k$.But obviously $s_m=m+1=f_m\lt f_{m+k}$.

Hi! According to the paper of AutoDAN, the time cost of each iter of AutoDAN is less than GCG. But as shown in your paper of EasyJailbreak, AutoDAN cost much...

https://github.com/lucidrains/PaLM-rlhf-pytorch/blob/6b02ee329106baff78e293afa7d1d2e6dd4e5ca2/palm_rlhf_pytorch/utils.py#L60 Using the sorted indices to index the sorted indices does not make sense. I think it may be `return logits.scatter(1, sorted_indices, sorted_logits)`