model.forward should have labels argument if you have loss return, otherwise during prediction_loop will cause length error.


it needs to add input_ids in dataset(for wav2vecProcessor, input will be input_values, I have to add an extra input_ids field XD)