Qilei Chen

Results 1 issues of Qilei Chen

### System Info Hi, there, When input_ids is more than 2 dimentions , the input_shape = input_ids.size() will give more than 2 numbers. Then bsz, seq_len = input_shape will crash....