haoneng.lhn
2023-05-31 9ee67cfbea90a89e44aaa9dfef7290548b1bbddd
funasr/layers/label_aggregation.py
@@ -79,4 +79,4 @@
        else:
            olens = None
        return output, olens
        return output.to(input.dtype), olens