Hi @Mrtranducdung ,
Finally, I reproduced the same problem as yours. It turns out the root cause is in UNQ_C1, which is one_step_attention. Please check following code snippet to make sure the order of its parameters.
2 Likes