shixian.shi
2023-03-13 9c21bbb96b95980ac059df991e442c437a69c828
funasr/layers/label_aggregation.py
@@ -79,4 +79,4 @@
        else:
            olens = None
        return output, olens
        return output.to(input.dtype), olens