funasr/layers/label_aggregation.py @@ -79,4 +79,4 @@ else: olens = None return output, olens return output.to(input.dtype), olens